clinic/netforce_clinic/models/je.py

import time
import xlrd

from netforce.model import Model, fields, get_model
from netforce.access import get_active_user, set_active_user
from netforce.utils import get_file_path
from netforce.utils import get_data_path

class JE(Model):
    _name="clinic.je"
    _string="Import Journal Entry"

    _fields={
        "number": fields.Char("Number"),
        "name": fields.Char("Description"),
        "date_import": fields.DateTime("Date Import"),
        "type": fields.Selection([("mg","Medical Government"),("sc","Social Security"),("nhso","NHSO (30฿)")],"Type"),
        'file': fields.File("File"),
        'payment_id': fields.Many2One("account.payment","Payment"),
        'lines': fields.One2Many("clinic.input.line","je_id", "Input Line"),
        'input_data': fields.One2Many("clinic.input.data","je_id", "Input Data"),
        'sheets': fields.One2Many("clinic.file.sheet","je_id", "Sheets"),
        'columns': fields.One2Many("clinic.file.column","je_id", "Columns"),
        "state": fields.Selection([("draft","Draft"),('sheet_loaded','Sheet Loaded'),("fail","Fail"),("done","Done")],"Status",required=True),
        'description': fields.Text("Description")
    }

    def _get_number(self,context={}):
        while 1:
            seq_id=get_model("sequence").find_sequence(name="Clinic Import JE")
            num=get_model("sequence").get_next_number(seq_id,context=context)
            if not num:
                return None
            user_id=get_active_user()
            set_active_user(1)
            res=self.search([["number","=",num]])
            set_active_user(user_id)
            if not res:
                return num
            get_model("sequence").increment_number(seq_id,context=context)
    
    def _get_name(self,context={}):
        timenow=time.strftime("%Y-%m-%d")
        return 'Import Journal Entry - %s'%timenow

    _defaults={
        'date_import': lambda *a: time.strftime("%Y-%m-%d %H:%M:%S"),
        'number': _get_number,
        'name': _get_name,
        'state': 'draft',
    }
    
    def load_sheet(self,ids,context={}):
        obj=self.browse(ids[0])
        fname=obj.file
        if not fname:
            raise Exception("Please select file!")
        fpath=get_file_path(fname)
        suffix=fpath.split(".")[-1]
        if suffix not in ('xls', 'xlsx'):
            raise Exception("ERROR : file support only xls, xlsx")

        wb=xlrd.open_workbook(fpath)
        sheets=wb.sheet_names()
        index=0
        # XXX clear old sheet
        for sheet in obj.sheets:
            sheet.delete()

        vals={
            'sheets': [],
        }

        for sheet in sheets:
            line={
                'index': index,
                'name': sheet,
            }
            vals["sheets"].append(("create",line))
        vals['state']='sheet_loaded'
        obj.write(vals)
        return {
            'next': {
                'name': "clinic_je",
                'mode': 'form',
                'active_id': obj.id,
            },
            'flash': "Load sheet successfully",
        }
    
    def load_data(self,ids,context={}):
        obj=self.browse(ids[0])
        if not obj.file:
            raise Exception("Please select some file!")
            
        if not obj.sheets:
            raise Exception("Please load sheet!")
        indexes=[]
        for sheet in obj.sheets:
            if sheet.select=='yes':
                indexes.append((sheet.index,sheet.name))
        if not indexes:
            raise Exception("Please select some sheet!")
        if len(indexes) > 1:
            raise Exception("Only one sheet can select!")
        sheet_name=indexes[0][1]    
        fname=obj.file
        fpath=get_file_path(fname)
        wb=xlrd.open_workbook(fpath)
        ws=wb.sheet_by_name(sheet_name)
        num_rows = ws.nrows - 1
        curr_row = -1
        records=[]
        import pprint
        while curr_row < num_rows:
            curr_row += 1
            #row=ws.row(curr_row)
            record={} 
            # XXX
            for col in (1,2,3,4):
                record[col]=ws.cell_value(curr_row, col)
            records.append(record)
        pprint.pprint(records)

    def find_invoice(self,ids,context={}):
        pass

    def import2journal(self,ids,context={}):
        pass
    
    def onchange_sheet(self,context={}):
        data=context["data"]
        path=context["path"]
        line=get_data_path(data,path,parent=True)
        for sheet in data['sheets']:
            if sheet['name']!=line['name']:
                sheet['select']=None
        return data

JE.register()