import logging from datetime import datetime, timedelta from odoo import models, fields, api 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) name = fields.Char(string='工作日历名称', required=True, size=15, length=30) working_shift_ids = fields.Many2many('sf.working.shift', string='班次', required=True) start_time = fields.Datetime(string='日开始时间', readonly=True, compute='_compute_working_shift_ids') end_time = fields.Datetime(string='日结束时间', readonly=True, compute='_compute_working_shift_ids') duration = fields.Char(string='时长', readonly=True, compute='_compute_working_shift_ids') day_off_ids = fields.Many2many('sf.day.off', string='休息日', required=True) status = fields.Selection([('正常', '正常'), ('禁用', '禁用')], string='状态', default='正常') 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.depends('working_shift_ids') def _compute_working_shift_ids(self): """ 根据所选班次自动生成开始时间和结束时间，同时计算出工作时长 :return: """ for record in self: if record: for working_shift_id in record.working_shift_ids: if not record.start_time: record.start_time = working_shift_id.start_time record.end_time = working_shift_id.end_time else: if (working_shift_id.start_time - record.start_time).total_seconds() < 0: record.start_time = working_shift_id.start_time if (working_shift_id.end_time - record.end_time).total_seconds() > 0: record.end_time = working_shift_id.end_time record.duration = record.end_time - record.start_time # @api.onchange('day_off_ids') # def _onchange_day_off_ids(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}) @api.model def create(self, vals): """ 创建记录时，生成工作日历 """ record = super(WorkLogSetting, self).create(vals) record.generate_work_calendar() return record def write(self, vals): """ 更新记录时，生成工作日历 """ # 先删除之前创建的工作日历事件记录 self.env['sf.work.schedule.calendar'].search([('calendar_code', '=', self.code)]).unlink() result = super(WorkLogSetting, self).write(vals) self.generate_work_calendar() return result def generate_work_calendar(self): """ 生成工作日历 """ self.ensure_one() # start_date = date.today() # 开始日期 # end_date = start_date + timedelta(days=365) # 结束日期 # 获取本年第一天和最后一天 start_date = datetime.now().replace(month=1, day=1).date() end_date = datetime.now().replace(month=12, day=31).date() # 休息日列表 rest_days = self.chinese_weekdays_to_english(self.day_off_ids.mapped('name')) for single_date in self.daterange(start_date, end_date): is_workday = self.chinese_weekday_to_english(single_date.strftime("%A")) logging.info(f"每天的星期：{is_workday}") if is_workday in rest_days: print('is_workday in rest_days', is_workday) self.env['sf.work.schedule.calendar'].sudo().create({ 'name': '休息日', 'name_id': self.id, 'date_time': single_date}) @staticmethod def chinese_weekdays_to_english(chinese_weekdays): """ 将元组的中文星期转换成英文 :param chinese_weekdays: :return: """ weekdays = { '星期一': 'Monday', '星期二': 'Tuesday', '星期三': 'Wednesday', '星期四': 'Thursday', '星期五': 'Friday', '星期六': 'Saturday', '星期日': 'Sunday' } english_weekdays = [] for chinese_weekday in chinese_weekdays: english_weekday = weekdays.get(chinese_weekday, 'Invalid weekday') english_weekdays.append(english_weekday) return english_weekdays @staticmethod def chinese_weekday_to_english(chinese_weekday): """ 将单个的中文星期转换成英文 :param chinese_weekday: :return: """ weekdays = { '星期一': 'Monday', '星期二': 'Tuesday', '星期三': 'Wednesday', '星期四': 'Thursday', '星期五': 'Friday', '星期六': 'Saturday', '星期日': 'Sunday' } weekday = weekdays.get(chinese_weekday) if weekday: return weekday return chinese_weekday @staticmethod def daterange(start_date, end_date): """ 生成日期范围 """ # 生成日期范围 for n in range(int((end_date - start_date).days)): yield start_date + timedelta(n) def open_work_schedule_calendar(self): action = self.env.ref('sf_plan_management.sf_work_schedule_calendar_act') result = action.read()[0] result['domain'] = [('name_id', '=', self.id)] return result 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([ ('Monday ', '星期一'), ('Tuesday ', '星期二'), ('Wednesday ', '星期三'), ('Thursday ', '星期四'), ('Friday ', '星期五'), ('Saturday ', '星期六'), ('Sunday ', '星期日')], '休息日名称') 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('工作日历编码', readonly=True, compute='_compute_name_id') 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) @api.depends('name_id') def _compute_name_id(self): for record in self: if record: record.calendar_code = record.name_id.code