Files
zhijie-ERP/selfhelp/models.py
T

331 lines
14 KiB
Python

# coding=utf-8
import datetime
import os
import xlrd
import decimal
from django.db import transaction
from django.db import models
from django.contrib.auth.models import User
from django.utils.text import force_text
from django.utils.translation import ugettext_lazy as _
from common import generic
from common import const
from mis import settings
from basedata.models import Material,ExtraParam,Project,ExpenseAccount,Measure
from organ.models import OrgUnit
class WorkOrder(generic.BO):
"""
"""
index_weight = 1
code = models.CharField(_("workorder code"),blank=True,null=True,max_length=const.DB_CHAR_CODE_10)
refer = models.ForeignKey("self",verbose_name=_("refer wo"),blank=True,null=True)
title = models.CharField(_("title"),max_length=const.DB_CHAR_NAME_120)
description = models.TextField(_("description"),blank=True,null=True)
business_domain = models.CharField(_("business domain"),max_length=const.DB_CHAR_CODE_4,choices=const.get_value_list('S045'),default='OT')
classification = models.CharField(_("classification"),max_length=const.DB_CHAR_CODE_4,choices=const.get_value_list('S044'),default='D')
service = models.ForeignKey(Material,verbose_name=_("service name"),null=True,blank=True,limit_choices_to={"is_virtual":"1"})
project = models.ForeignKey(Project,verbose_name=_("project"),null=True,blank=True)
status = models.CharField(_("status"),blank=True,null=True,default='NEW',max_length=const.DB_CHAR_CODE_6,choices=const.get_value_list('S046'))
answer = models.TextField(_("answer"),blank=True,null=True)
user = models.ForeignKey(User,verbose_name=_("user"),blank=True,null=True)
attach = models.FileField(_('attach'),blank=True,null=True,help_text=u'工单附件,不导入明细。')
detail = models.FileField(_('to be imported detail'),blank=True,null=True,help_text=u'您可导入需求明细,模板请参考文档FD0007')
def __unicode__(self):
return u"%s-%s" % (self.code,self.title)
def save(self, force_insert=False, force_update=False, using=None,
update_fields=None):
super(WorkOrder,self).save(force_insert,force_update,using,update_fields)
if self.service:
material = self.service
if self.woextravalue_set.count() < 1 and material.extraparam_set and material.extraparam_set.count() > 0:
for param in material.extraparam_set.all():
extra_param = WOExtraValue.objects.create(workorder=self,param_name=param)
self.woextravalue_set.add(extra_param)
item_count = WOItem.objects.filter(workorder=self).count()
if self.detail and item_count == 0:
path = os.path.join(settings.MEDIA_ROOT,self.detail.name)
workbook = xlrd.open_workbook(path)
sheet = workbook.sheet_by_index(0)
row_count = sheet.nrows
with transaction.atomic():
for row_index in range(row_count):
row = sheet.row_values(row_index)
if row_index == 0:
doc_type = row[1]
if doc_type.startswith('0'):
break
else:
continue
elif row_index < 3:
continue
material = None
measure = None
try:
measure = Measure.objects.get(code=row[4])
except Exception,e:
measure = Measure.objects.create(code=row[4],name=force_text(row[5]))
try:
material = Material.objects.get(code=row[0])
except Exception,e:
material = Material(code=row[0],name=force_text(row[1]),spec=force_text(row[2]))
material.save()
WOItem.objects.create(workorder=self,material=material,measure=measure,amount=row[6])
class Meta:
verbose_name = _("workorder apply")
verbose_name_plural = _("workorder apply")
class Media:
js = ('js/workorder.js',)
class WOExtraValue(models.Model):
"""
"""
workorder = models.ForeignKey(WorkOrder,verbose_name=_("workorder"))
param_name = models.ForeignKey(ExtraParam,verbose_name=_("extra param"))
param_value = models.CharField(_("param value"),blank=True,null=True,max_length=const.DB_CHAR_NAME_40)
class Meta:
verbose_name = _("extra value")
verbose_name_plural = _("extra values")
class WOItem(models.Model):
"""
"""
workorder = models.ForeignKey(WorkOrder,verbose_name=_("workorder"))
material = models.ForeignKey(Material,verbose_name=_("material"),null=True,blank=True,limit_choices_to={"is_virtual":"0"})
amount = models.DecimalField(_("amount"),max_digits=10,decimal_places=4,blank=True,null=True)
measure = models.ForeignKey(Measure,verbose_name=_('measure'),blank=True,null=True)
price = models.DecimalField(_("price"),max_digits=10,decimal_places=4,blank=True,null=True)
class Meta:
verbose_name = _("workorder item")
verbose_name_plural = _("workorder items")
class Loan(generic.BO):
"""
"""
LOAD_STATUS = (
('N',_("NEW")),
('I',_("IN PROGRESS")),
('A',_("APPROVED")),
('P',_("PAYED"))
)
index_weight = 3
code = models.CharField(_("loan code"),max_length=const.DB_CHAR_CODE_10,blank=True,null=True)
title = models.CharField(_("title"),max_length=const.DB_CHAR_NAME_120)
description = models.TextField(_("description"),blank=True,null=True)
project = models.ForeignKey(Project,verbose_name=_("project"))
user = models.ForeignKey(User,verbose_name=_("user"),blank=True,null=True)
status = models.CharField(_("status"),blank=True,null=True,default='N',max_length=const.DB_CHAR_CODE_2,choices=LOAD_STATUS)
logout_time = models.DateTimeField(_("logout time"),blank=True,null=True)
loan_amount = models.DecimalField(_("loan amount"),max_digits=10,decimal_places=2,blank=True,null=True)
logout_amount = models.DecimalField(_("logout amount"),max_digits=10,decimal_places=2,blank=True,null=True)
pay_user = models.CharField(_('pay user'),max_length=const.DB_CHAR_NAME_40,blank=True,null=True)
pay_time = models.DateTimeField(_('pay time'),blank=True,null=True)
is_clear = models.BooleanField(_('is clear'),default=False)
def __unicode__(self):
import decimal
left = self.loan_amount
left -= self.logout_amount or decimal.Decimal(0.00)
name = '%s%s'%(self.user.last_name,self.user.first_name)
return '%s %s %s J:%.2f Y:%.2f' % (self.code,name,self.title,self.loan_amount,left)
def applier(self):
return u'%s%s'%(self.user.last_name,self.user.first_name)
applier.short_description = _('applier')
def action_pay(self,request):
self.pay_time = datetime.datetime.now()
self.pay_user = request.user.username
self.status = 'P'
self.save()
class Meta:
verbose_name = _("loan")
verbose_name_plural = _("loans")
permissions = (
('financial_pay',_("financial pay")),
)
class Reimbursement(generic.BO):
"""
"""
REIM_STATUS = (
('N',_("NEW")),
('I',_("IN PROGRESS")),
('A',_("APPROVED")),
('P',_("PAYED"))
)
index_weight = 2
code = models.CharField(_("fee code"),max_length=const.DB_CHAR_CODE_10,blank=True,null=True)
title = models.CharField(_("title"),max_length=const.DB_CHAR_NAME_120)
description = models.TextField(_("description"),blank=True,null=True)
project = models.ForeignKey(Project,verbose_name=_("project"))
wo = models.ForeignKey(WorkOrder,verbose_name=_("work order"),null=True,blank=True)
user = models.ForeignKey(User,verbose_name=_("user"),blank=True,null=True)
org = models.ForeignKey(OrgUnit,verbose_name=_("cost center"),blank=True,null=True)
bank_account = models.CharField(_("bank account"),max_length=const.DB_CHAR_NAME_120,blank=True,null=True)
status = models.CharField(_("status"),blank=True,null=True,default='N',max_length=const.DB_CHAR_CODE_2,choices=REIM_STATUS)
amount = models.DecimalField(_("amount of money"),max_digits=10,decimal_places=2,blank=True,null=True)
loan = models.ForeignKey(Loan,verbose_name=_("loan record"),blank=True,null=True)
loan_amount = models.DecimalField(_("loan amount"),max_digits=10,decimal_places=2,blank=True,null=True)
logout_amount = models.DecimalField(_("logout amount"),max_digits=10,decimal_places=2,blank=True,null=True)
pay_amount = models.DecimalField(_("pay amount"),max_digits=10,decimal_places=2,blank=True,null=True)
pay_time = models.DateTimeField(_("pay time"),blank=True,null=True)
pay_user = models.CharField(_('pay user'),max_length=const.DB_CHAR_NAME_40,blank=True,null=True)
def applier(self):
return u'%s%s'%(self.user.last_name,self.user.first_name)
def save(self, force_insert=False, force_update=False, using=None,
update_fields=None):
import decimal
if self.loan:
left = self.loan.loan_amount
left -= self.loan.logout_amount or decimal.Decimal(0.00)
if self.logout_amount is None or self.logout_amount == '':
if self.amount < left:
self.logout_amount = self.amount
else:
self.logout_amount = left
self.pay_amount = self.amount
if self.logout_amount:
self.pay_amount -= self.logout_amount
super(Reimbursement,self).save(force_insert,force_update,using,update_fields)
def action_pay(self,request):
if self.loan:
if self.logout_amount is None or self.logout_amount == '':
self.logout_amount = self.amount
if self.logout_amount < 0:
raise Exception(u'核销金额小于0')
if self.loan.logout_amount is None:
self.loan.logout_amount = self.logout_amount
else:
self.loan.logout_amount += self.logout_amount
if self.loan.loan_amount == self.loan.logout_amount:
self.loan.is_clear=True
self.loan.logout_time = datetime.datetime.now()
self.loan.save()
if self.amount > self.logout_amount:
self.pay_amount = self.amount - self.logout_amount
self.pay_time = datetime.datetime.now()
self.pay_user = request.user.username
self.status = 'P'
self.save()
applier.short_description = _('applier')
class Meta:
verbose_name = _("reimbursement")
verbose_name_plural = _("reimbursements")
permissions = (
('financial_pay',_("financial pay")),
)
class ReimbursementItem(models.Model):
"""
"""
import datetime
reimbursement = models.ForeignKey(Reimbursement,verbose_name=_("reimbursement"))
expense_account = models.ForeignKey(ExpenseAccount,verbose_name=_("expenses account"))
begin = models.DateField(_("occur date"),default=datetime.date.today)
amount = models.DecimalField(_("amount of money"),max_digits=10,decimal_places=2)
memo = models.CharField(_("memo"),max_length=const.DB_CHAR_NAME_40,blank=True,null=True)
def save(self, force_insert=False, force_update=False, using=None,
update_fields=None):
super(ReimbursementItem,self).save(force_insert,force_update,using,update_fields)
sql = 'UPDATE selfhelp_reimbursement SET amount = (SELECT SUM(amount) FROM selfhelp_reimbursementitem WHERE reimbursement_id = %s) WHERE id = %s'
params = [self.reimbursement.id,self.reimbursement.id]
generic.update(sql,params)
class Meta:
verbose_name = _("fee item")
verbose_name_plural = _("fee items")
class Activity(generic.BO):
"""
"""
CLASSIFICATION = (
('T',_("Train")),
('M',_("Meeting")),
('G',_("Community")),
)
index_index_weight = 4
begin_time = models.DateTimeField(_('begin time'))
end_time = models.DateTimeField(_('end time'))
enroll_deadline = models.DateTimeField(_('enroll deadline'),blank=True,null=True)
code = models.CharField(_("code"),max_length=const.DB_CHAR_NAME_20,blank=True,null=True)
title = models.CharField(_("title"),max_length=const.DB_CHAR_NAME_120)
parent = models.ForeignKey('self',verbose_name=_("parent"),blank=True,null=True)
description = models.TextField(_("description"),blank=True,null=True)
host = models.CharField(_("host"),max_length=const.DB_CHAR_NAME_80,blank=True,null=True)
speaker = models.CharField(_("speaker"),max_length=const.DB_CHAR_NAME_80,blank=True,null=True)
accept_enroll = models.BooleanField(_("accept enroll"),default=1)
room = models.ForeignKey(Material,verbose_name=_("room"),blank=True,null=True,limit_choices_to={'tp':20})
location = models.CharField(_("location"),max_length=const.DB_CHAR_NAME_80,blank=True,null=True)
classification = models.CharField(_("classification"),max_length=const.DB_CHAR_CODE_2,blank=True,null=True,choices=CLASSIFICATION,default='M')
mail_list = models.TextField(_("mail list"),blank=True,null=True)
mail_notice = models.BooleanField(_("mail notice"),default=1)
short_message_notice = models.BooleanField(_("short message notice"),default=1)
weixin_notice = models.BooleanField(_("weixin notice"),default=1)
status = models.BooleanField(_("published"),default=0)
publish_time = models.DateTimeField(_("publish time"),blank=True,null=True)
attach = models.FileField(_("attach"),blank=True,null=True,upload_to='activity')
class Meta:
verbose_name = _("activity")
verbose_name_plural = _("activities")
class Feedback(models.Model):
"""
"""
RANK = (
('A','A'),
('B','B'),
('C','C'),
('D','D'),
)
activity = models.ForeignKey(Activity)
user = models.ForeignKey(User)
feed_time = models.DateTimeField(_("feedback time"),auto_now_add=True)
rank = models.CharField(_('rank'),max_length=const.DB_CHAR_CODE_2,blank=True,null=True,choices=RANK,default='B')
comment = models.CharField(_("suggest"),blank=True,null=True,max_length=const.DB_CHAR_NAME_80)
class Enroll(models.Model):
"""
"""
activity = models.ForeignKey(Activity)
user = models.ForeignKey(User)
enroll_time = models.DateTimeField(_("enroll time"),auto_now_add=True)