kanq-erp 0.1.0__tar.gz

kanq_erp-0.1.0/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: kanq_erp
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: drissionpage
+Requires-Dist: python-dotenv>=1.2.1
+Requires-Dist: typer>=0.21.1

kanq_erp-0.1.0/kanq_erp/__init__.py

@@ -0,0 +1,32 @@
+"""
+@File    : __init__.py.py
+@Time    : 2026/1/27 上午11:57
+@Author  : xiashuobad
+@Desc    : 
+
+/**
+ * _ooOoo_
+ * o8888888o
+ * 88" . "88
+ * (| -_- |)
+ *  O\ = /O
+ * ___/`---'\____
+ * .   ' \\| |// `.
+ * / \\||| : |||// \
+ * / _||||| -:- |||||- \
+ * | | \\\ - /// | |
+ * | \_| ''\---/'' | |
+ * \ .-\__ `-` ___/-. /
+ * ___`. .' /--.--\ `. . __
+ * ."" '< `.___\_<|>_/___.' >'"".
+ * | | : `- \`.;`\ _ /`;.`/ - ` : | |
+ * \ \ `-. \_ __\ /__ _/ .-` / /
+ * ======`-.____`-.___\_____/___.-`____.-'======
+ * `=---=' bug泛滥 佛已瘫痪
+"""
+from .erp import app
+from .browser_manager import BrowserManager
+from .day_report import DayReportHandler
+from .element_utils import ElementUtils
+from .llm_summary import LLMSummarizer
+from .week_plan import WeekPlanHandler

kanq_erp-0.1.0/kanq_erp/browser_manager.py

@@ -0,0 +1,51 @@
+"""
+Author: xiash$
+Date: 2026/1/26$
+Project: week_plan$
+Description: $
+***
+* _ooOoo_
+* o8888888o
+* 88" . "88
+* (| -_- |)
+*  O\ = /O
+* ___/`---'\____
+* .   ' \\| |// `.
+* / \\||| : |||// \
+* / _||||| -:- |||||- \
+* | | \\\ - /// | |
+* | \_| ''\---/'' | |
+* \ .-\__ `-` ___/-. /
+* ___`. .' /--.--\ `. . __
+* ."" '< `.___\_<|>_/___.' >'"".
+* | | : `- \`.;`\ _ /`;.`/ - ` : | |
+* \ \ `-. \_ __\ /__ _/ .-` / /
+* ======`-.____`-.___\_____/___.-`____.-'======
+* `=---='
+*          .............................................
+*           佛曰：bug泛滥，我已瘫痪！
+"""
+
+# browser_manager.py
+from DrissionPage import Chromium, ChromiumOptions
+
+
+class BrowserManager:
+    def __init__(self, browser_path=None):
+        self.browser_path = browser_path
+        self.tab = None
+
+    def initialize(self):
+        co = ChromiumOptions()
+        self.browser_path and co.set_browser_path(self.browser_path)
+        self.tab = Chromium(co).new_tab()
+        return self.tab
+
+    def login(self, url, username, password):
+        self.tab.get(url)
+        ele_username = self.tab.ele('#username', timeout=5)
+        if ele_username:
+            ele_username.input(username)
+            self.tab.ele('#password').input(password)
+            self.tab.ele('#submit').click()
+        return ele_username, self.tab.ele('#homeid')

kanq_erp-0.1.0/kanq_erp/day_report.py

@@ -0,0 +1,77 @@
+# day_report.py
+from .element_utils import ElementUtils
+
+
+class DayReportHandler:
+    def __init__(self, browser_manager):
+        self.browser = browser_manager
+        self.tab = browser_manager.tab
+
+    def navigate_to_day_report(self):
+        ElementUtils.safe_click(self.tab, '@class=workReport-true')
+
+    def select_task(self, task_name):
+        row = self.tab.ele(f'xpath://table//tr[td="{task_name}"]')
+        self.tab.actions.move_to(row).click(times=2)
+
+        return float(row.eles('tag:td')[4].text.strip())
+
+    def get_progress_value(self, task_name):
+        form = self.tab.ele('@name=任务汇报明细')
+        row = form.ele(f'xpath://table//tr[td="{task_name}"]')
+        return float(row.eles('tag:td')[7].text.strip() or 0)
+
+    def open_report_form(self):
+        span = self.tab.ele('@title=任务汇报')
+        self.tab.actions.move_to(span).click()
+
+    def fill_report_form(
+            self,
+            description: str,
+            completion_rate: float,
+            date: str = None,
+            location: str = '武汉技术中心',
+            task_type: str = '需求开发',
+            quality: str = '优',
+            workload: float = 8.00
+    ):
+        ElementUtils.safe_input(self.tab, '#65_text', date)
+        ElementUtils.safe_input(self.tab, '@name=工作地点', location)
+        ElementUtils.safe_input(self.tab, '@name=任务类型_text', task_type)
+
+        dropdown = self.tab.ele('@class=co-combo-down')
+        dropdown.click()
+        option = self.tab.wait.ele_displayed(f'text={quality}', timeout=3)
+        option.click.left(by_js=True)
+
+        ElementUtils.safe_input(self.tab, '@name=当日投入工作量', f"{workload:.2f}")
+        ElementUtils.safe_input(self.tab, '@name=完成率', f'{completion_rate:.2f}')
+        ElementUtils.safe_input(self.tab, '@name=工作内容描述', description)
+
+    def submit_report(self):
+        confirm_button = self.tab.wait.ele_displayed('@class=co-win-button co-alert-button', timeout=1)
+        if confirm_button:
+            confirm_button.click.left(by_js=True)
+        ElementUtils.safe_click(self.tab, '@onclick=33-6-4')
+        print('报工完成！')
+
+# def main():
+#     browser = BrowserManager()
+#     browser.initialize()
+#     browser.login('http://erp.kqgeo.com:889/cops/home/index.jsp', 'xiangsong', 'Xs135456')
+#
+#     handler = DayReportHandler(browser)
+#     handler.navigate_to_day_report()
+#     handler.select_task('解译平台功能研发与维护')
+#     progress_value = handler.get_progress_value('解译平台功能研发与维护')
+#     handler.open_report_form()
+#
+#     handler.fill_report_form(
+#         completion_rate=progress_value + 0.1,
+#         description='666666666666666666666666666666666666666666666666'
+#     )
+#     handler.submit_report()
+#
+#
+# if __name__ == '__main__':
+#     main()

kanq_erp-0.1.0/kanq_erp/element_utils.py

@@ -0,0 +1,53 @@
+"""
+Author: xiash$
+Date: 2026/1/26$
+Project: week_plan$
+Description: $
+***
+* _ooOoo_
+* o8888888o
+* 88" . "88
+* (| -_- |)
+*  O\ = /O
+* ___/`---'\____
+* .   ' \\| |// `.
+* / \\||| : |||// \
+* / _||||| -:- |||||- \
+* | | \\\ - /// | |
+* | \_| ''\---/'' | |
+* \ .-\__ `-` ___/-. /
+* ___`. .' /--.--\ `. . __
+* ."" '< `.___\_<|>_/___.' >'"".
+* | | : `- \`.;`\ _ /`;.`/ - ` : | |
+* \ \ `-. \_ __\ /__ _/ .-` / /
+* ======`-.____`-.___\_____/___.-`____.-'======
+* `=---='
+*          .............................................
+*           佛曰：bug泛滥，我已瘫痪！
+"""
+import time
+
+
+# element_utils.py
+class ElementUtils:
+    @staticmethod
+    def safe_input(tab, selector, text, clear=True):
+        element = tab.ele(selector)
+        if element and text:
+            element.input(text, clear=clear)
+
+    @staticmethod
+    def safe_click(tab, selector, timeout=10):
+        element = tab.wait.ele_displayed(selector, timeout=timeout)
+        if element:
+            time.sleep(1)
+            element.click()
+
+    @staticmethod
+    def safe_get_text(tab, selector, default=''):
+        """等待元素显示并获取文本"""
+        element = tab.wait.ele_displayed(selector)
+        if element:
+            text = element.property('value').strip()
+            return text if text else default
+        return default

kanq_erp-0.1.0/kanq_erp/erp.py

@@ -0,0 +1,173 @@
+"""
+Author: xiash$
+Date: 2026/1/26$
+Project: week_plan$
+Description: $
+***
+* _ooOoo_
+* o8888888o
+* 88" . "88
+* (| -_- |)
+*  O\ = /O
+* ___/`---'\____
+* .   ' \\| |// `.
+* / \\||| : |||// \
+* / _||||| -:- |||||- \
+* | | \\\ - /// | |
+* | \_| ''\---/'' | |
+* \ .-\__ `-` ___/-. /
+* ___`. .' /--.--\ `. . __
+* ."" '< `.___\_<|>_/___.' >'"".
+* | | : `- \`.;`\ _ /`;.`/ - ` : | |
+* \ \ `-. \_ __\ /__ _/ .-` / /
+* ======`-.____`-.___\_____/___.-`____.-'======
+* `=---='
+*          .............................................
+*           佛曰：bug泛滥，我已瘫痪！
+"""
+import typer
+import os
+import sys
+from dotenv import load_dotenv, set_key
+from pathlib import Path
+from DrissionPage.errors import ElementNotFoundError
+
+from .browser_manager import BrowserManager
+
+app = typer.Typer()
+
+# 环境变量配置文件路径
+ENV_FILE = Path.home() / ".erp_env"
+
+
+def ensure_env_file():
+    """确保环境变量文件存在"""
+    if not ENV_FILE.exists():
+        ENV_FILE.touch()
+        set_key(ENV_FILE, 'ERP_URL', 'http://erp.kqgeo.com:889/cops/home/index.jsp')
+
+
+def save_env(key, value=None):
+    """保存环境变量"""
+    if value is not None:
+        set_key(ENV_FILE, key, value)
+    else:
+        set_key(ENV_FILE, key, typer.prompt(f'请输入 {key}'))
+    # 重新加载环境变量
+    load_dotenv(ENV_FILE)
+
+
+def get_env_value(key, prompt=True):
+    """获取环境变量值"""
+    load_dotenv(ENV_FILE)
+    value = os.environ.get(key, None)
+    if value is None and prompt:
+        value = typer.prompt(f"请输入 {key.lower()}")
+        save_env(key, value)
+
+    return value
+
+
+def retry_login(browser: BrowserManager, erp_url, username, password, try_times=3):
+    for i in range(try_times):
+        login_flag, index_flag = browser.login(erp_url, username, password)
+        if index_flag:
+            break
+        elif i == try_times - 1:
+            sys.exit(1)
+        elif not login_flag and not index_flag:
+            erp_url = typer.prompt('erp地址有误！请输入新的erp地址')
+            set_key(ENV_FILE, 'ERP_URL', erp_url)
+        else:
+            print('用户或密码错误！')
+            username = typer.prompt(f'请输入用户名')
+            password = typer.prompt('请输入密码')
+            set_key(ENV_FILE, 'ERP_USERNAME', username)
+            set_key(ENV_FILE, 'ERP_PASSWORD', password)
+
+
+def login_erp(erp_url=None, browser_path=None):
+    ensure_env_file()
+    if browser_path:
+        set_key(ENV_FILE, 'BROWSER_PATH', browser_path)
+    browser = BrowserManager(get_env_value('BROWSER_PATH', prompt=False))
+    try:
+        browser.initialize()
+    except FileNotFoundError:
+        print('无法自动找到浏览器程序，请通过 --浏览器 参数手动指定程序路径（一次指定，永久更新）')
+        sys.exit(1)
+
+    if erp_url:
+        set_key(ENV_FILE, 'ERP_URL', erp_url)
+    erp_url = get_env_value('ERP_URL')
+    username = get_env_value('ERP_USERNAME')
+    password = get_env_value('ERP_PASSWORD')
+    retry_login(browser, erp_url, username, password)
+    return browser
+
+
+def validate_report_length(text: str):
+    if len(text) < 15:
+        raise typer.BadParameter("工作内容描述不得少于15字！")
+    return text
+
+
+def all_task_type():
+    return ['需求开发', '系统设计', '系统搭建', '编码实现', '数据处理', '系统测试', '系统缺陷修改', '项目验收',
+            '技术文档编写', '项目管理', '技术评审', '会议', '技术培训', '其他', '售前演示交流', '方案撰写']
+
+
+@app.command('报工')
+def day_report(
+        text: str = typer.Argument(None, callback=validate_report_length, help='报工内容'),
+        # complete_rate: float = typer.Option(None, '--进度', help='项目完成进度', min=0, max=100),
+        task_name: str = typer.Option(None, '--任务', help='任务名称'),
+        date: str = typer.Option(None, '--日期', help='日期，格式：2026-01-26，默认为当天'),
+        location: str = typer.Option('武汉技术中心', '--地点', help='报工地点'),
+        task_type: str = typer.Option('需求开发', '--类型', autocompletion=all_task_type, help='任务类型'),
+        workload: float = typer.Option(8., '--时长', help='报工时长', min=0.5),
+        erp_url: str = typer.Option(None, '--网址', help='erp网址，有默认值，当网址发生变化时，可设置更新'),
+        browser_path: str = typer.Option(None, '--浏览器', help='浏览器程序exe文件所在路径,默认使用谷歌浏览器')
+):
+    from .day_report import DayReportHandler
+
+    browser = login_erp(erp_url, browser_path)
+    handler = DayReportHandler(browser)
+    handler.navigate_to_day_report()
+    if task_name:
+        set_key(ENV_FILE, 'TASK_NAME', task_name)
+    try:
+        total_complete_rate = handler.select_task(get_env_value('TASK_NAME'))
+        progress_value = handler.get_progress_value(get_env_value('TASK_NAME'))
+    except ElementNotFoundError:
+        print('任务名称有误！请通过参数 --任务 设置更新要报工的任务（只需设置一次），名称不要用单引号！')
+        sys.exit(1)
+    handler.open_report_form()
+
+    handler.fill_report_form(
+        completion_rate=min(max(progress_value, total_complete_rate) + 0.1, 100.),
+        description=text,
+        date=date,
+        location=location,
+        task_type=task_type,
+        workload=workload
+    )
+    handler.submit_report()
+
+
+@app.command('周计划')
+def week_plan(
+        erp_url: str = typer.Option(None, '--网址', help='erp网址，有默认值，当网址发生变化时，可设置更新'),
+        browser_path: str = typer.Option(None, '--浏览器', help='浏览器程序exe文件所在路径,默认使用谷歌浏览器')
+):
+    from .week_plan import WeekPlanHandler
+
+    browser = login_erp(erp_url, browser_path)
+
+    handler = WeekPlanHandler(browser)
+    handler.navigate_to_week_plan()
+    all_reports = handler.collect_daily_reports()
+    week_report = handler.generate_week_summary(all_reports)
+    next_week_plan = handler.generate_next_week_plan(week_report)
+    handler.fill_week_reports(week_report, next_week_plan)
+    handler.save_report()

kanq_erp-0.1.0/kanq_erp/llm_summary.py

@@ -0,0 +1,99 @@
+from typing import Optional
+from DrissionPage import Chromium, ChromiumOptions
+from DrissionPage._functions.keys import Keys
+
+LLM_URL = 'https://www.qianwen.com/'
+DEFAULT_PROMPT = ("帮我总结一下上面的报工，按项目名称分类,项目名称中关键字重复度高的视为同一个项目，"
+                  "直接给出结果，返回格式为：1.项目1：。。。一段话；2.项目2：。。。一段话；3.项目3：。。。。，"
+                  "注意：报工中每行冒号前面就是项目名，结果中请不要出现报工中没有的项目名！！！")
+
+
+class LLMSummarizer:
+    """LLM摘要生成器类，用于使用千问大模型生成文本摘要，支持上下文管理器"""
+
+    def __init__(self, headless: bool = True, browser_path: Optional[str] = None):
+        """
+        初始化LLM摘要生成器
+
+        Args:
+            headless: 是否使用无头模式，默认为True
+            browser_path: 浏览器可执行文件路径，如果为None则使用默认路径
+        """
+        self.headless = headless
+        self.browser_path = browser_path
+        self.tab = None
+
+    def initialize(self) -> None:
+        """初始化浏览器和标签页"""
+        co = ChromiumOptions()
+        if self.headless:
+            co.headless()
+        if self.browser_path:
+            co.set_browser_path(self.browser_path)
+
+        self.tab = Chromium(co).new_tab()
+        self.tab.get(LLM_URL)
+
+    def summarize(self, text: str, prompt: Optional[str] = None) -> str:
+        """
+        生成文本摘要
+
+        Args:
+            text: 需要摘要的文本
+            prompt: 提示词，如果为None则使用默认提示词
+
+        Returns:
+            str: 生成的摘要文本
+        """
+        if not self.tab:
+            self.initialize()
+
+        # 如果没有提供提示词，使用默认提示词
+        if not prompt:
+            prompt = DEFAULT_PROMPT
+
+        # 构建完整的输入文本
+        contents = f"{text}\n\n{prompt}{Keys.ENTER}"
+
+        # 输入文本到文本框
+        self.tab.ele('text=向千问提问').input(contents)
+
+        self.tab.ele('@class=answerToolsContent-pftvzO', timeout=30)
+
+        # 获取结果
+        qk_texts = '\n'.join(map(lambda span: span.text, self.tab.eles('@class=qk-md-text complete')))
+
+        return qk_texts
+
+    def close(self) -> None:
+        """关闭浏览器"""
+        if self.tab:
+            self.tab.close()
+            self.tab = None
+
+    def __enter__(self):
+        """支持上下文管理器，进入with语句块时调用"""
+        self.initialize()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """支持上下文管理器，退出with语句块时调用"""
+        self.close()
+
+
+# 使用示例
+if __name__ == "__main__":
+    # 使用with语句，自动管理资源
+    with LLMSummarizer(headless=False) as summarizer:
+        # 需要摘要的文本
+        text_to_summarize = """
+        BS_AI：继续编写tensorflow后端语义分割遥感影像的推理代码，已初步完成，单进程推理已通过测试，多卡多进程模式还有问题调试中。
+        BS_AI：继续调试tensorflow后端语义分割遥感影像的推理代码，逐步排查定位多卡多进程导致的报错问题，问题已找到并解决。
+        BS_AI：调试ogc推理代码，成功解决ogc推理报错的问题，主要修改有：wms 1.3.0版本下参数格式的修改，适配1.3.0之前及1.3.0的版本；接触瓦片最大数量的限制，根据geojson的范围去下载影像而不是下载全图，防止内存溢出；修复图像尺寸小于模型推理最小尺寸时报错的bug
+        BS_AI：开始编写tensorflow后端目标检测训练的代码，已完成：训练数据加载器dataset的封装；dfine算法模型的构建与封装；数据增强管道的封装。
+        BS_AI：调试ogc推理时结果为空的问题，定位问题为rasterio保存的ogc图片未自动刷新缓存导致的结果为空，已修复；继续编写调试tensorflow后端目标检测训练的接口，新增对LinearLR和MultiStepLR学习率调度器的封装，新增mAP指标类的封装。
+        """
+
+        # 生成摘要
+        summary = summarizer.summarize(text_to_summarize)
+        print(summary)

kanq_erp-0.1.0/kanq_erp/week_plan.py

@@ -0,0 +1,67 @@
+# week_plan.py
+from .element_utils import ElementUtils
+from .llm_summary import LLMSummarizer
+
+
+class WeekPlanHandler:
+    def __init__(self, browser_manager):
+        self.browser = browser_manager
+        self.tab = browser_manager.tab
+        self.week_days = ['Mon', 'Tues', 'Wed', 'Thur', 'Fri']
+
+    def navigate_to_week_plan(self):
+        ElementUtils.safe_click(self.tab, '#weekly_a')
+        ElementUtils.safe_click(self.tab, 'text=提取日志/报工')
+
+    def collect_daily_reports(self):
+        all_reports = []
+        for day in self.week_days:
+            day_report = ElementUtils.safe_get_text(self.tab, f'#thisWeek{day}_M', '请假')
+            ElementUtils.safe_input(self.tab, f'#thisWeek{day}_M', day_report)
+            ElementUtils.safe_input(self.tab, f'#thisWeek{day}_A', day_report)
+            all_reports.append(day_report)
+        return all_reports
+
+    @staticmethod
+    def generate_week_summary(all_reports):
+        with LLMSummarizer() as summarizer:
+            return summarizer.summarize('\n'.join(all_reports))
+
+    @staticmethod
+    def generate_next_week_plan(week_report):
+        with LLMSummarizer() as summarizer:
+            return summarizer.summarize(
+                week_report,
+                prompt='这是这周的报工内容，帮我生成一下下周计划，直接给出至少十条计划，其他废话不要有！'
+            )
+
+    def fill_week_reports(self, week_report, next_week_plan):
+        ElementUtils.safe_input(self.tab, '#thisWeekPoint', week_report)
+        ElementUtils.safe_input(self.tab, '#nextWeekPoint', next_week_plan)
+
+        plan_items = next_week_plan.split('\n')
+        for i, day in enumerate(self.week_days):
+            i *= 2
+            ElementUtils.safe_input(self.tab, f'#nextWeek{day}_M', plan_items[i])
+            ElementUtils.safe_input(self.tab, f'#nextWeek{day}_A', plan_items[i + 1])
+
+    def save_report(self):
+        ElementUtils.safe_click(self.tab, 'text=保存')
+        print('周计划已保存！')
+
+# def main():
+#     browser = BrowserManager()
+#     browser.initialize()
+#     browser.login('http://erp.kqgeo.com:889/cops/home/index.jsp', 'xiangsong', 'Xs135456')
+#
+#     handler = WeekPlanHandler(browser)
+#     handler.navigate_to_week_plan()
+#     all_reports = handler.collect_daily_reports()
+#     week_report = handler.generate_week_summary(all_reports)
+#     next_week_plan = handler.generate_next_week_plan(week_report)
+#     handler.fill_week_reports(week_report, next_week_plan)
+#     handler.save_report()
+#
+#
+# if __name__ == '__main__':
+#     main()

kanq_erp-0.1.0/kanq_erp.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: kanq_erp
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: drissionpage
+Requires-Dist: python-dotenv>=1.2.1
+Requires-Dist: typer>=0.21.1

kanq_erp-0.1.0/kanq_erp.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+README.md
+pyproject.toml
+kanq_erp/__init__.py
+kanq_erp/browser_manager.py
+kanq_erp/day_report.py
+kanq_erp/element_utils.py
+kanq_erp/erp.py
+kanq_erp/llm_summary.py
+kanq_erp/week_plan.py
+kanq_erp.egg-info/PKG-INFO
+kanq_erp.egg-info/SOURCES.txt
+kanq_erp.egg-info/dependency_links.txt
+kanq_erp.egg-info/entry_points.txt
+kanq_erp.egg-info/requires.txt
+kanq_erp.egg-info/top_level.txt

kanq_erp-0.1.0/kanq_erp.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

kanq_erp-0.1.0/kanq_erp.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+erp = kanq_erp.erp:app

kanq_erp-0.1.0/kanq_erp.egg-info/requires.txt

@@ -0,0 +1,3 @@
+drissionpage
+python-dotenv>=1.2.1
+typer>=0.21.1

kanq_erp-0.1.0/kanq_erp.egg-info/top_level.txt

@@ -0,0 +1 @@
+kanq_erp

kanq_erp-0.1.0/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "kanq_erp"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    'drissionpage',
+    "python-dotenv>=1.2.1",
+    "typer>=0.21.1",
+]
+
+[project.scripts]
+erp = "kanq_erp.erp:app"
+
+[tool.setuptools]
+packages = ['kanq_erp']

kanq_erp-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+