from datetime import datetime, timedelta from odoo import models, fields, api import re def time_H_selection(): return [('00', '00'), ('01', '01'), ('02', '02'), ('03', '03'), ('04', '04'), ('05', '05'), ('06', '06'), ('07', '07'), ('08', '08'), ('09', '09'), ('10', '10'), ('11', '11'), ('12', '12'), ('13', '13'), ('14', '14'), ('15', '15'), ('16', '16'), ('17', '17'), ('18', '18'), ('19', '19'), ('20', '20'), ('21', '21'), ('22', '22'), ('23', '23')] def time_M_or_S_selection(): return [('00', '00'), ('05', '05'), ('10', '10'), ('15', '15'), ('20', '20'), ('25', '25'), ('30', '30'), ('35', '35'), ('40', '40'), ('45', '45'), ('50', '50'), ('55', '55')] class WorkLogSetting(models.Model): _name = 'sf.work.log.setting' _description = '工作日历设置' def _get_code(self): """ 自动生成编码 :return: """ fixture_material = self.env['sf.work.log.setting'].sudo().search( [('code', '!=', '')], limit=1, order="id desc") if not fixture_material: num = "%03d" % 1 else: m = int(fixture_material.code) + 1 num = "%03d" % m return num code = fields.Char(string='序号', default=_get_code, readonly=True) name = fields.Char(string='工作日历名称', required=True) start_time = fields.Char(string='日开始时间', readonly=True, compute='_compute_start_time') start_time_H = fields.Selection(time_H_selection(), '时', required=True) start_time_M = fields.Selection(time_M_or_S_selection(), '分', required=True) end_time = fields.Char(string='日结束时间', readonly=True, compute='_compute_end_time') end_time_H = fields.Selection(time_H_selection(), '时', required=True) end_time_M = fields.Selection(time_M_or_S_selection(), '分', required=True) duration = fields.Char(string='时长', readonly=True, compute='_compute_duration') day_off_id = fields.Many2many('sf.day.off', string='休息日', required=True) working_shift_id = fields.Many2many('sf.working.shift', string='班次') status = fields.Boolean(string='状态', default=True) update_person = fields.Char(string='更新人', default=lambda self: self.env.user.name) update_time = fields.Datetime(string='更新时间', default=lambda self: fields.Datetime.now()) setting_to_calendar_ids = fields.One2many('sf.work.schedule.calendar', 'name_id', '工作日历') # @api.model # def create(self, vals): # vals['setting_to_calendar_ids'] = [(4, child.id)] # return super(WorkLogSetting, self).create(vals) @api.depends('start_time_H', 'start_time_M') def _compute_start_time(self): """ 设置输入日开始时间 :return: """ for record in self: record.start_time = f"{record.start_time_H}:{record.start_time_M}:00" @api.depends('end_time_H', 'end_time_M') def _compute_end_time(self): """ 设置输入日结束时间 :return: """ for record in self: record.end_time = f"{record.end_time_H}:{record.end_time_M}:00" @api.depends('start_time_H', 'start_time_M', 'end_time_H', 'end_time_M') def _compute_duration(self): """ 根据日开始时间和日结束时间计算每日工作时长 :return: """ for record in self: st_h = float(record.start_time_H) st_m = float(record.start_time_M) end_h = float(record.end_time_H) end_m = float(record.end_time_M) # 日开始时间小于日结束时间 if st_h < end_h: record.duration = str(round(end_h - st_h + (end_m - st_m) / 60, 2)) else: record.duration = str(round(end_h - st_h + (end_m - st_m) / 60 + 24, 2)) @api.onchange('day_off_id') def _onchange_day_off_id(self): # 先删除之前创建的工作日历事件记录 # self.env['sf.work.schedule.calendar'].search([('calendar_code', '=', self.code)]).unlink() # 获取当年的一月一号的日期 year = fields.Datetime.now().year # 2023 first_day = datetime(year, 1, 1).date() # 2023-01-01 day_of_week = first_day.strftime("%A") # 星期日 # 根据day_of_week，将其设置为起始0，循环周一到周日，按循环顺序设置为0-6 # 列：{'星期日': 0, '星期一': 1, '星期二': 2, '星期三': 3, '星期四': 4, '星期五': 5, '星期六': 6} desc = {} desc_weekdays = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期日'] status = False ints = 0 while len(desc) < 7: if len(desc) != 0: for week in desc_weekdays: if len(desc) < 7: desc.update({week: ints}) ints += 1 else: for week in desc_weekdays: if status: desc.update({week: ints}) ints += 1 if day_of_week == week: desc.update({week: ints}) status = True ints += 1 # 创建工作日历的休息日事件 day_off_ids = self.day_off_id.mapped('name') record = self.env['sf.work.log.setting'].search([('code', '=', self.code)]) # record.setting_to_calendar_ids.unlink() # self.env.cr.commit() for week in day_off_ids: num = desc[week] while num <= 30: target_date = first_day + timedelta(days=num) num += 7 self.env['sf.work.schedule.calendar'].create({'name': '休息日', 'name_id': record.id, 'calendar_code': self.code, 'date_time': target_date}) class WorkingShift(models.Model): _name = 'sf.working.shift' _description = '班次' def _get_code(self): """ 自动生成编码 :return: """ fixture_material = self.env['sf.working.shift'].sudo().search( [('code', '!=', '')], limit=1, order="id desc") if not fixture_material: num = "%03d" % 1 else: m = int(fixture_material.code) + 1 num = "%03d" % m return num code = fields.Char('编码', default=_get_code, readonly=True) name = fields.Char('名称', required=True) start_time = fields.Datetime('班次开始时间') end_time = fields.Datetime('班次结束时间') remark = fields.Char('备注') class DayOff(models.Model): _name = 'sf.day.off' _description = '休息日' name = fields.Selection([ ('星期一', '星期一'), ('星期二', '星期二'), ('星期三', '星期三'), ('星期四', '星期四'), ('星期五', '星期五'), ('星期六', '星期六'), ('星期日', '星期日')], '休息日名称') class WorkScheduleCalendar(models.Model): _name = 'sf.work.schedule.calendar' _description = '工作日历' name = fields.Selection([('休息日', '休息日'), ('计划停机', '计划停机')], '日历事件名称') date_time = fields.Date('休息时间') name_id = fields.Many2one('sf.work.log.setting', '工作日历名称') calendar_code = fields.Char('工作日历编码') day_off_id = fields.Many2many('sf.day.off', string='休息日') scheduled_outage = fields.Char('计划停机') monthly_rest_days = fields.Char('月休息天数', readonly=True) annual_rest_days = fields.Char('年休息天数', readonly=True) monthly_planned_downtime = fields.Char('月计划停机时长', readonly=True) annual_planned_downtime = fields.Char('年计划停机时长', readonly=True)