134 lines
5.0 KiB
Python
134 lines
5.0 KiB
Python
import time
|
|
import datetime
|
|
|
|
from docx import Document
|
|
from docx.shared import Inches
|
|
|
|
import win32com
|
|
from win32com.client import Dispatch, constants
|
|
import os
|
|
import sys
|
|
import ruamel.yaml as Yaml
|
|
import logging.config
|
|
|
|
|
|
class WEEK_REPORT:
|
|
file_name = ""
|
|
pre_file_name = ""
|
|
pre_str = ""
|
|
str = ""
|
|
user_name = ""
|
|
project_name = ""
|
|
project_leader = ""
|
|
__word = ""
|
|
__doc = ""
|
|
|
|
def __init__(self):
|
|
with open("config.yaml", "r", encoding="UTF-8") as f1:
|
|
yaml = Yaml.YAML(typ='safe', pure=True)
|
|
data = yaml.load(f1)
|
|
logger.debug("config.yaml: {}".format(data))
|
|
self.user_name = data["user_name"]
|
|
self.project_name = data["project_name"]
|
|
self.project_leader = data["project_leader"]
|
|
self.get_name()
|
|
|
|
def get_name(self):
|
|
self.file_name = "{}WW{}周工作报告_{}.doc".format(time.strftime('%Y'), time.strftime('%W'), self.user_name)
|
|
if time.strftime('%W') == "01":
|
|
self.pre_file_name = "{}WW{}周工作报告_{}.doc".format(str(int(time.strftime('%Y')) - 1), str(52),
|
|
self.user_name)
|
|
else:
|
|
self.pre_file_name = "{}WW{}周工作报告_{}.doc".format(time.strftime('%Y'),
|
|
str(int(time.strftime('%W')) - 1), self.user_name)
|
|
logger.info("pre file name: {}".format(self.pre_file_name))
|
|
logger.info("new file name: {};".format(self.file_name))
|
|
if os.path.isfile(self.file_name):
|
|
logger.info("{} is already exist.".format(self.file_name))
|
|
self.open_file()
|
|
sys.exit(1)
|
|
|
|
def copy_file(self):
|
|
os.system('copy {} {}'.format(self.pre_file_name, self.file_name))
|
|
|
|
def open_file(self):
|
|
self.word = Dispatch('Word.Application')
|
|
# 或者使用下面的方法，使用启动独立的进程：
|
|
# word = DispatchEx('Word.Application')
|
|
self.word.Visible = 1 # 0:后台运行 1:前台运行(可见)
|
|
self.word.DisplayAlerts = 0 # 不显示，不警告
|
|
self.doc = self.word.Documents.Open(os.getcwd() + "\\{}".format(self.file_name))
|
|
|
|
def edit_file(self):
|
|
table = self.doc.Tables(1)
|
|
|
|
# 工作周
|
|
table.Cell(Row=1, Column=2).Range.Text = "{}WW{}".format(time.strftime('%Y'), time.strftime('%W'))
|
|
|
|
# 日期
|
|
self.get_last_week()
|
|
self.get_current_week()
|
|
table.Cell(Row=1, Column=4).Range.Text = self.str
|
|
|
|
# 所在项目
|
|
table.Cell(Row=2, Column=2).Range.Text = self.project_name
|
|
|
|
# 项目负责人
|
|
table.Cell(Row=2, Column=4).Range.Text = self.project_leader
|
|
|
|
# 删除上周工作内容
|
|
for i in range(7,12):
|
|
logger.info("process {} row ...".format(i))
|
|
table.Cell(Row=i, Column=2).Range.Text = ''
|
|
table.Cell(Row=i, Column=3).Range.Text = ''
|
|
table.Cell(Row=13, Column=1).Range.Text = ''
|
|
table.Cell(Row=15, Column=1).Range.Text = '项目工作：\n'
|
|
|
|
def update_file(self):
|
|
self.open_file()
|
|
self.edit_file()
|
|
self.doc.Save()
|
|
# doc.SaveAs(os.getcwd() + "\\funOpenExistFile.docx") # 另存为
|
|
|
|
def close_file(self):
|
|
self.doc.Close() # 关闭 word 文档
|
|
self.word.Quit() # 关闭 office
|
|
|
|
def get_last_week(self, date=None):
|
|
if date:
|
|
today = datetime.datetime.strptime(str(date), '%Y-%m-%d')
|
|
else:
|
|
today = datetime.datetime.today()
|
|
end_time = today - datetime.timedelta(days=today.isoweekday()) - datetime.timedelta(days=2)
|
|
start_time = end_time - datetime.timedelta(days=4)
|
|
# return start_time.strftime("%Y-%m-%d"), end_time.strftime("%Y-%m-%d")
|
|
self.pre_str = start_time.strftime("%#m.%#d") + "~" + end_time.strftime("%#m.%#d")
|
|
logger.debug("pre_str: {}".format(self.pre_str))
|
|
|
|
def get_current_week(self, date=None):
|
|
if date:
|
|
duty_date = datetime.datetime.strptime(str(date), '%Y-%m-%d')
|
|
monday, friday = duty_date, duty_date
|
|
else:
|
|
monday, friday = datetime.date.today(), datetime.date.today()
|
|
one_day = datetime.timedelta(days=1)
|
|
while monday.weekday() != 0:
|
|
monday -= one_day
|
|
while friday.weekday() != 4:
|
|
friday += one_day
|
|
# return datetime.datetime.strftime(monday, "%Y-%m-%d"), datetime.datetime.strftime(friday, "%Y-%m-%d")
|
|
self.str = datetime.datetime.strftime(monday, "%#m.%#d") + "~" + datetime.datetime.strftime(friday, "%#m.%#d")
|
|
logger.debug("str: {}".format(self.str))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with open("log_config.yaml", 'r') as f:
|
|
yaml = Yaml.YAML(typ='safe', pure=True)
|
|
config = yaml.load(f.read())
|
|
logging.config.dictConfig(config)
|
|
logger = logging.getLogger("file_logger")
|
|
|
|
week_report = WEEK_REPORT()
|
|
week_report.copy_file()
|
|
week_report.update_file()
|