61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
import time
|
|
from calendar import monthrange
|
|
|
|
from netforce.model import Model,fields,get_model
|
|
|
|
|
|
class ComputeLaborCost(Model):
|
|
_name="clinic.compute.labor.cost"
|
|
_string="Compute Labor Cost"
|
|
_transient=True
|
|
|
|
_fields={
|
|
"date": fields.Date("Month"),
|
|
"date_from": fields.Date("From", required=True),
|
|
"date_to": fields.Date("To", required=True),
|
|
}
|
|
|
|
def _get_date_from(self,context={}):
|
|
year,month=time.strftime("%Y-%m").split("-")
|
|
return '%s-%s-01'%(year,month)
|
|
|
|
def _get_date_to(self,context={}):
|
|
year,month,day=time.strftime("%Y-%m-%d").split("-")
|
|
weekday, total_day=monthrange(int(year), int(month))
|
|
return "%s-%s-%s"%(year,month,total_day)
|
|
|
|
_defaults={
|
|
'date': lambda *a: time.strftime("%Y-%m-%d"),
|
|
'date_from': _get_date_from,
|
|
'date_to': _get_date_to,
|
|
}
|
|
|
|
def compute(self,ids,context={}):
|
|
obj=self.browse(ids)[0]
|
|
dom=[]
|
|
dom.append(['date','>=', obj.date_from])
|
|
dom.append(['date','<=', obj.date_to])
|
|
for lcost in get_model("clinic.labor.cost").search_browse(dom):
|
|
print("compute %s ....", lcost.cycle_item_id.name)
|
|
lcost.compute()
|
|
print("Done!")
|
|
return {
|
|
'next': {
|
|
'name': 'clinic_compute_labor_cost',
|
|
'mode': 'form',
|
|
'active_id': obj.id,
|
|
},
|
|
'flash': 'Compute Successfully',
|
|
}
|
|
|
|
def onchange_date(self,context={}):
|
|
data=context['data']
|
|
date=data['date']
|
|
year,month,day=date.split("-")
|
|
weekday, total_day=monthrange(int(year), int(month))
|
|
data['date_from']="%s-%s-01"%(year,month)
|
|
data['date_to']="%s-%s-%s"%(year,month,total_day)
|
|
return data
|
|
|
|
ComputeLaborCost.register()
|