zjz-cli 0.0.1__tar.gz

zjz_cli-0.0.1/PKG-INFO

@@ -0,0 +1,65 @@
+Metadata-Version: 2.4
+Name: zjz_cli
+Version: 0.0.1
+Summary: 自记账的工具箱
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: annotated-doc>=0.0.4
+Requires-Dist: anyio>=4.0.0
+Requires-Dist: certifi>=2024.2.25
+Requires-Dist: click>=8.3.1
+Requires-Dist: h11>=0.16.0
+Requires-Dist: httpcore>=1.0.9
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: idna>=3.0
+Requires-Dist: markdown-it-py>=4.0.0
+Requires-Dist: mdurl>=0.1.2
+Requires-Dist: packaging>=24.0
+Requires-Dist: Pygments>=2.10.0
+Requires-Dist: rich>=14.3.3
+Requires-Dist: shellingham>=1.5.4
+Requires-Dist: typer>=0.24.1
+Requires-Dist: typing_extensions>=4.12.0
+
+## zjz_cli
+
+Cli tool for zjz
+
+### Use VENV
+
+```shell
+mkdir .venv
+cd .venv
+python3.12 -m venv .
+source .venv/bin/activate
+```
+
+### Install dependence
+
+```shell
+pip install -r requirements.txt
+```
+
+### Build
+
+```shell
+pyinstaller --onefile --name zjz_cli --clean --noupx main.py
+```
+
+### Dependence
+
+If you add dependence, add here
+
+```shell
+pip install typer
+pip install pyinstaller
+pip install httpx
+```
+
+### Export requirements
+
+If you add dependence, export it
+
+```shell
+pip freeze > requirements.txt
+```

zjz_cli-0.0.1/README.md

@@ -0,0 +1,42 @@
+## zjz_cli
+
+Cli tool for zjz
+
+### Use VENV
+
+```shell
+mkdir .venv
+cd .venv
+python3.12 -m venv .
+source .venv/bin/activate
+```
+
+### Install dependence
+
+```shell
+pip install -r requirements.txt
+```
+
+### Build
+
+```shell
+pyinstaller --onefile --name zjz_cli --clean --noupx main.py
+```
+
+### Dependence
+
+If you add dependence, add here
+
+```shell
+pip install typer
+pip install pyinstaller
+pip install httpx
+```
+
+### Export requirements
+
+If you add dependence, export it
+
+```shell
+pip freeze > requirements.txt
+```

zjz_cli-0.0.1/app/api/zjz_api.py

@@ -0,0 +1,184 @@
+from app.api.zjz_req import zjz_req
+
+
+def sms_verify_code(phone: str) -> dict:
+    return zjz_req.get_client().post_json('login/sms', {
+        'phone': phone,
+    })
+
+
+def verify_sms_code(phone: str, code: str) -> dict:
+    return zjz_req.get_client().post_json('login/sms-2-token', {
+        'phone': phone,
+        'sms': code,
+    })
+
+
+def get_employee_list(uid: str = None, keyword: str = '', page: int = 1) -> dict:
+    if not page or page < 1:
+        page = 1
+    return zjz_req.get_client().get('ryxx', params={
+        'uid': uid,
+        'keyword': keyword,
+        'page': page
+    })
+
+
+def add_employee(uid: str, data: dict) -> dict:
+    data['uid'] = uid
+    return zjz_req.get_client().post_json('ryxx', data)
+
+
+def update_employee(employee_id: str, data: dict) -> dict:
+    return zjz_req.get_client().put('ryxx', {
+        'id': employee_id,
+        'data': data,
+    })
+
+
+def get_payroll(uid: str, year: int, month: int) -> dict:
+    month_str = f"{year}-{month:02d}"
+    return zjz_req.get_client().get('payroll', params={
+        'uid': uid,
+        'month': month_str
+    })
+
+
+def get_payroll_creation_status(uid: str, year: int, month: int) -> dict:
+    month_str = f"{year}-{month:02d}"
+    return zjz_req.get_client().get('payroll/creation-status', params={
+        'uid': uid,
+        'month': month_str
+    })
+
+
+def get_preview_payroll(uid: str, year: int, month: int) -> dict:
+    month_str = f"{year}-{month:02d}"
+    return zjz_req.get_client().get('payroll/preview', params={
+        'uid': uid,
+        'month': month_str
+    })
+
+
+def create_payroll(uid: str, year: int, month: int, data: list) -> dict:
+    month_str = f"{year}-{month:02d}"
+    return zjz_req.get_client().post_json('payroll', {
+        'uid': uid,
+        'month': month_str,
+        'data': data
+    })
+
+
+def get_sales_invoice_data_list(uid: str, current: int = 1, invoice_number: str = '', invoice_date_begin: str = '', invoice_date_end: str = '', status: str = '') -> dict:
+    return zjz_req.get_client().get('invoice/sales_invoice_data_list', params={
+        'uid': uid,
+        'current': current,
+        'invoice_number': invoice_number,
+        'invoice_date_begin': invoice_date_begin,
+        'invoice_date_end': invoice_date_end,
+        'status': status,
+    })
+
+
+def set_month_no_sales_invoice(uid: str, year: int, month: int) -> dict:
+    return zjz_req.get_client().post_json('invoice/set_month_no_sales_invoice', {
+        'uid': uid,
+        'year': year,
+        'month': month,
+    })
+
+
+def get_income_invoice_data_list(uid: str, current: int = 1, invoice_number: str = '', invoice_date_begin: str = '', invoice_date_end: str = '', status: str = '') -> dict:
+    return zjz_req.get_client().get('invoice/income_invoice_data_list', params={
+        'uid': uid,
+        'current': current,
+        'invoice_number': invoice_number,
+        'invoice_date_begin': invoice_date_begin,
+        'invoice_date_end': invoice_date_end,
+        'status': status,
+    })
+
+
+def set_month_no_invoice(uid: str, year: int, month: int) -> dict:
+    return zjz_req.get_client().post_json('invoice/set_month_no_invoice', {
+        'uid': uid,
+        'year': year,
+        'month': month,
+    })
+
+
+def invoice_can_choose_use_list(uid: str, goods_name: str, total_tax_price: str, total_money: str) -> dict:
+    return zjz_req.get_client().get('invoice/invoice_can_choose_use_list', params={
+        'uid': uid,
+        'goods_name': goods_name,
+        'total_tax_price': total_tax_price,
+        'total_money': total_money,
+    })
+
+
+def invoice_choose_use(uid: str, invoice_id: int, category: str, is_pay: str, payee: str) -> dict:
+    return zjz_req.get_client().post_json('invoice/invoice_choose_use', {
+        'uid': uid,
+        'invoice_id': invoice_id,
+        'category': category,
+        'is_pay': is_pay,
+        'payee': payee,
+    })
+
+
+def get_todo_list(uid: str = '') -> dict:
+    params = {}
+    if uid:
+        params['uid'] = uid
+    return zjz_req.get_client().get('todo-list', params=params)
+
+
+def before_set_month_audited(year: int, month: int, uid: str = '') -> dict:
+    params = {
+        'year': year,
+        'month': month,
+    }
+    if uid:
+        params['uid'] = uid
+    return zjz_req.get_client().get('audit/before_set_month_audited', params=params)
+
+
+def set_month_audited(year: int, month: int, uid: str = '') -> dict:
+    payload = {
+        'year': year,
+        'month': month,
+    }
+    if uid:
+        payload['uid'] = uid
+    return zjz_req.get_client().post_json('audit/set_month_audited', payload)
+
+
+def get_bank_bill_data_list(uid: str, current: int = 1, date_begin: str = '', date_end: str = '') -> dict:
+    params = {
+        'current': current,
+        'date_begin': date_begin,
+        'date_end': date_end,
+    }
+    if uid:
+        params['uid'] = uid
+    return zjz_req.get_client().get('bank/bill_data_list', params=params)
+
+
+def choose_bank_use(uid: str, bill_id: int, rule_text: str) -> dict:
+    payload = {
+        'id': bill_id,
+        'rule_text': rule_text,
+    }
+    if uid:
+        payload['uid'] = uid
+    return zjz_req.get_client().post_json('bank/bank_bill_choose_use', payload)
+
+
+def set_month_bank_no_water(uid: str, year: int, month: int) -> dict:
+    payload = {
+        'year': year,
+        'month': month,
+    }
+    if uid:
+        payload['uid'] = uid
+    return zjz_req.get_client().post_json('bank/set_month_no_water', payload)

zjz_cli-0.0.1/app/api/zjz_req.py

@@ -0,0 +1,22 @@
+from app.service.config_manager import cfg_mgr
+from app.util.http_client import HttpClient
+
+
+class ZjzReq:
+    def __init__(self):
+        self.headers = {}
+        self.base_url: str = "https://www.zijizhang.com/ai/skill"
+        self.client = None
+
+    def get_client(self):
+        auth_token = self.headers.get('Authorization')
+        if not auth_token:
+            token = cfg_mgr.get_config('token')
+            if token:
+                self.headers['Authorization'] = f'Bearer {token}'
+                self.client = None
+        if not self.client:
+            self.client = HttpClient(base_url=self.base_url, headers=self.headers)
+        return self.client
+
+zjz_req = ZjzReq()

zjz_cli-0.0.1/app/module/account.py

@@ -0,0 +1,47 @@
+import typer
+
+from app.service.account_service import AccountService
+from app.util import echo_util
+
+account_app = typer.Typer()
+
+
+@account_app.command(
+    name='get_sms_verify_code',
+    help='获取登录的短信验证码',
+    no_args_is_help=True)
+def get_sms_verify_code(phone: str = typer.Argument(..., help='自记账的登录手机号')):
+    service = AccountService()
+    res = service.sms_verify_code(phone)
+    echo_util.echo_http_res(res)
+
+
+@account_app.command(
+    name='verify_sms_code',
+    help='通过短信验证码登录',
+    no_args_is_help=True)
+def verify_sms_code(
+        phone: str = typer.Argument(..., help='手机号'),
+        code: str = typer.Argument(..., help='短信验证码')):
+    service = AccountService()
+    res = service.sms_code_login(phone, code)
+    echo_util.echo_http_res(res)
+
+
+@account_app.command(
+    name='get_switchable_users',
+    help='获取可切换的公司列表')
+def get_switchable_users():
+    service = AccountService()
+    res = service.get_switchable_user()
+    echo_util.echo_http_res(res)
+
+
+@account_app.command(
+    name='switch_to',
+    help='切换公司',
+    no_args_is_help=True)
+def switch_to(uid: str = typer.Argument(..., help='公司唯一标识uid')):
+    service = AccountService()
+    res = service.switch_to(uid)
+    echo_util.echo_http_res(res)

zjz_cli-0.0.1/app/module/bank.py

@@ -0,0 +1,49 @@
+import typer
+
+from app.service.bank_service import BankService
+from app.util import echo_util
+
+bank_app = typer.Typer()
+
+
+@bank_app.command(
+    name='get_bank_bill_data_list',
+    help="获取银行回单/流水列表")
+def get_bank_bill_data_list(
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+    current: int = typer.Option(1, '--current', help='页码，默认 1'),
+    date_begin: str = typer.Option('', '--date_begin', help='交易日期开始（YYYY-MM-DD；如需使用需与 date_end 同时传）'),
+    date_end: str = typer.Option('', '--date_end', help='交易日期结束（YYYY-MM-DD；如需使用需与 date_begin 同时传）'),
+):
+    service = BankService()
+    res = service.get_bank_bill_data_list(uid, current, date_begin, date_end)
+    echo_util.echo_http_res(res)
+
+
+@bank_app.command(
+    name='choose_bank_use',
+    help="回单选择用途/入账",
+    no_args_is_help=True)
+def choose_bank_use(
+    bill_id: int = typer.Argument(...,help='回单单据 id（来自 get_bank_bill_data_list 返回 bill_list[].id）'),
+    rule_text: str = typer.Argument(..., help='用途文本（建议从 bill_list[].use_list 中选择）'),
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+):
+    service = BankService()
+    res = service.choose_bank_use(uid, bill_id, rule_text)
+    echo_util.echo_http_res(res)
+
+
+@bank_app.command(
+    name='set_month_bank_no_water',
+    help="标记某个月份无银行回单/流水",
+    no_args_is_help=True)
+def set_month_bank_no_water(
+    year: int = typer.Argument(..., help='年份，如 2026'),
+    month: int = typer.Argument(..., help='月份(1-12)，如 3'),
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+):
+    service = BankService()
+    res = service.set_month_bank_no_water(uid, year, month)
+    echo_util.echo_http_res(res)
+

zjz_cli-0.0.1/app/module/employee.py

@@ -0,0 +1,48 @@
+import json
+import typer
+
+from app.service.employee_service import EmployeeService
+from app.util import echo_util
+
+employee_app = typer.Typer()
+
+
+@employee_app.command(
+    name='get_employee_list',
+    help="获取员工列表",
+    epilog="""使用示例\n\nzjz_cli employee get_employee_list""")
+def get_employee_list(
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid'),
+        keyword: str = typer.Option('', '--keyword', help='姓名/证件号码 关键字，支持模糊搜索，如果为空则查询全部')
+):
+    service = EmployeeService()
+    res = service.get_employee_list(uid, keyword)
+    echo_util.echo_http_res(res)
+
+
+@employee_app.command(
+    name='add_employee',
+    help="添加员工",
+    no_args_is_help=True)
+def add_employee(
+        data: str = typer.Argument(..., help='员工信息的json字符串'),
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid')
+):
+    data_obj = json.loads(data)
+    service = EmployeeService()
+    res = service.add_employee(uid, data_obj)
+    echo_util.echo_http_res(res)
+
+
+@employee_app.command(
+    name='update_employee',
+    help='更新员工信息',
+    no_args_is_help=True)
+def update_employee(
+        employee_id: str = typer.Argument(..., help='员工id'),
+        data: str = typer.Argument(..., help='需要更新信息的json字符串')
+):
+    data_obj = json.loads(data)
+    service = EmployeeService()
+    res = service.update_employee(employee_id, data_obj)
+    echo_util.echo_http_res(res)

zjz_cli-0.0.1/app/module/invoice.py

@@ -0,0 +1,96 @@
+import typer
+
+from app.service.invoice_service import InvoiceService
+from app.util import echo_util
+
+invoice_app = typer.Typer()
+
+
+@invoice_app.command(
+    name='get_sales_invoice_data_list',
+    help="获取销项发票列表")
+def get_sales_invoice_data_list(
+        uid: str = typer.Option('', '--uid', help='用户uid'),
+        current: int = typer.Option(1, '--current', help='页码'),
+        invoice_number: str = typer.Option('', '--invoice_number', help='发票号码 关键字，指定发票号码查询，如果为空则查询全部'),
+        invoice_date_begin: str = typer.Option('', '--invoice_date_begin', help='发票日期开始 关键字，指定发票日期开始查询（如果要使用，invoice_date_begin和invoice_date_end两个都需要），如果为空则不限制，日期格式为YYYY-MM-DD，例如2024-01-01'),
+        invoice_date_end: str = typer.Option('', '--invoice_date_end', help='发票日期结束 关键字，指定发票日期结束查询（如果要使用，invoice_date_begin和invoice_date_end两个都需要），如果为空则不限制，日期格式为YYYY-MM-DD，例如2024-01-31'),
+):
+    service = InvoiceService()
+    res = service.get_sales_invoice_data_list(uid, current, invoice_number, invoice_date_begin, invoice_date_end)
+    echo_util.echo_http_res(res)
+
+
+@invoice_app.command(
+    name='get_income_invoice_data_list',
+    help="获取进项/收到的发票列表")
+def get_income_invoice_data_list(
+        uid: str = typer.Option('', '--uid', help='用户uid'),
+        current: int = typer.Option(1, '--current', help='页码'),
+        invoice_number: str = typer.Option('', '--invoice_number', help='发票号码关键字（精确匹配），不传则查询全部'),
+        invoice_date_begin: str = typer.Option('', '--invoice_date_begin', help='开票日期开始（YYYY-MM-DD；如需使用需与 invoice_date_end 同时传）'),
+        invoice_date_end: str = typer.Option('', '--invoice_date_end', help='开票日期结束（YYYY-MM-DD；如需使用需与 invoice_date_begin 同时传）'),
+        status: str = typer.Option('', '--status', help='发票状态（审核中/待确认/已拒绝/已入账），不传则查询全部'),
+):
+    service = InvoiceService()
+    res = service.get_income_invoice_data_list(uid, current, invoice_number, invoice_date_begin, invoice_date_end, status)
+    echo_util.echo_http_res(res)
+
+
+@invoice_app.command(
+    name='set_month_no_sales_invoice',
+    help="标记某个月份无销项/开出的发票",
+    no_args_is_help=True)
+def set_month_no_sales_invoice(
+    uid: str = typer.Option('', '--uid', help='用户uid'),
+    year: int = typer.Argument(..., help='年份，如 2026'),
+    month: int = typer.Argument(..., help='月份(1-12)，如 3'),
+):
+    service = InvoiceService()
+    res = service.set_month_no_sales_invoice(uid, year, month)
+    echo_util.echo_http_res(res)
+
+
+@invoice_app.command(
+    name='set_month_no_invoice',
+    help="标记某个月份无进项/收到的发票",
+    no_args_is_help=True)
+def set_month_no_invoice(
+    uid: str = typer.Option('', '--uid', help='用户uid'),
+    year: int = typer.Argument(..., help='年份，如 2026'),
+    month: int = typer.Argument(..., help='月份(1-12)，如 3'),
+):
+    service = InvoiceService()
+    res = service.set_month_no_invoice(uid, year, month)
+    echo_util.echo_http_res(res)
+
+
+@invoice_app.command(
+    name='invoice_can_choose_use_list',
+    help="进项发票：查询可选用途列表",
+    no_args_is_help=True)
+def invoice_can_choose_use_list(
+    uid: str = typer.Option('', '--uid', help='用户uid（不传则使用当前已切换在用的 uid）'),
+    goods_name: str = typer.Option(..., '--goods_name', help='货物/服务名称（用于推荐用途）'),
+    total_tax_price: str = typer.Option(..., '--total_tax_price', help='价税合计（建议传“元”，如 113.00）'),
+    total_money: str = typer.Option(..., '--total_money', help='不含税金额（建议传“元”，如 100.00）'),
+):
+    service = InvoiceService()
+    res = service.invoice_can_choose_use_list(uid, goods_name, total_tax_price, total_money)
+    echo_util.echo_http_res(res)
+
+
+@invoice_app.command(
+    name='invoice_choose_use',
+    help="进项发票：选择用途并入账",
+    no_args_is_help=True)
+def invoice_choose_use(
+    uid: str = typer.Option('', '--uid', help='用户uid（不传则使用当前已切换在用的 uid）'),
+    invoice_id: int = typer.Option(..., '--invoice_id', help='发票 id'),
+    category: str = typer.Option(..., '--category', help='用途/类目（来自 invoice_can_choose_use_list.items[].id'),
+    is_pay: str = typer.Option(..., '--is_pay', help='支付方式（对公银行支付/员工垫付/老板垫付/现金支付）'),
+    payee: str = typer.Option(..., '--payee', help='垫付人/收款人（服务端会按支付方式校验）'),
+):
+    service = InvoiceService()
+    res = service.invoice_choose_use(uid, invoice_id, category, is_pay, payee)
+    echo_util.echo_http_res(res)

zjz_cli-0.0.1/app/module/payroll.py

@@ -0,0 +1,79 @@
+import json
+import typer
+
+from app.service.payroll_service import PayrollService
+from app.util import echo_util
+
+payroll_app = typer.Typer()
+
+
+def parse_month(month_str: str) -> tuple:
+    try:
+        parts = month_str.split('-')
+        if len(parts) != 2:
+            raise typer.BadParameter("格式必须为 yyyy-mm，例如 2026-04")
+        year = int(parts[0])
+        month_val = int(parts[1])
+        if not (1 <= month_val <= 12):
+            raise typer.BadParameter("月份必须在 1 到 12 之间")
+        return year, month_val
+    except ValueError:
+        raise typer.BadParameter("年份和月份必须是数字，例如 2026-04")
+
+
+@payroll_app.command(
+    name='get_payroll',
+    help="获取工资单",
+    epilog="""使用示例\n\nzjz_cli payroll get_payroll 2026-04""")
+def get_payroll(
+        month: str = typer.Argument(..., help='工资单月份，如 2026-04'),
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid')
+):
+    req_year, req_month = parse_month(month)
+    service = PayrollService()
+    res = service.get_payroll(uid, req_year, req_month)
+    echo_util.echo_http_res(res)
+
+
+@payroll_app.command(
+    name='get_payroll_creation_status',
+    help="获取工资单创建状态",
+    no_args_is_help=True)
+def get_payroll_creation_status(
+        month: str = typer.Argument(..., help='按月时间，如 2026-04'),
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid')
+):
+    req_year, req_month = parse_month(month)
+    service = PayrollService()
+    res = service.get_payroll_creation_status(uid, req_year, req_month)
+    echo_util.echo_http_res(res)
+
+
+@payroll_app.command(
+    name='get_preview_payroll',
+    help="获取预览工资单",
+    no_args_is_help=True)
+def get_preview_payroll(
+        month: str = typer.Argument(..., help='工资单月份，如 2026-04'),
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid')
+):
+    req_year, req_month = parse_month(month)
+    service = PayrollService()
+    res = service.get_preview_payroll(uid, req_year, req_month)
+    echo_util.echo_http_res(res)
+
+
+@payroll_app.command(
+    name='create_payroll',
+    help='创建工资单',
+    no_args_is_help=True)
+def create_payroll(
+        month: str = typer.Argument(..., help='工资单月份，如 2026-04'),
+        data: str = typer.Argument(..., help='工资单列表的json字符串'),
+        uid: str = typer.Option('', '--uid', help='公司唯一标识uid')
+):
+    req_year, req_month = parse_month(month)
+    data_obj = json.loads(data)
+    service = PayrollService()
+    res = service.create_payroll(uid, req_year, req_month, data_obj)
+    echo_util.echo_http_res(res)

zjz_cli-0.0.1/app/module/todo.py

@@ -0,0 +1,45 @@
+import typer
+
+from app.service.todo_service import TodoService
+from app.util import echo_util
+
+todo_app = typer.Typer()
+
+
+@todo_app.command(
+    name='get_todo_list',
+    help="获取代办列表")
+def get_todo_list(
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+):
+    service = TodoService()
+    res = service.get_todo_list(uid)
+    echo_util.echo_http_res(res)
+
+
+@todo_app.command(
+    name='before_set_month_audited',
+    help="提交审账前确认",
+    no_args_is_help=True)
+def before_set_month_audited(
+    year: int = typer.Argument(..., help='年份，如 2026'),
+    month: int = typer.Argument(..., help='月份(1-12)，如 3'),
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+):
+    service = TodoService()
+    res = service.before_set_month_audited(year, month, uid)
+    echo_util.echo_http_res(res)
+
+
+@todo_app.command(
+    name='set_month_audited',
+    help="提交审账",
+    no_args_is_help=True)
+def set_month_audited(
+    year: int = typer.Argument(..., help='年份，如 2026'),
+    month: int = typer.Argument(..., help='月份(1-12)，如 3'),
+    uid: str = typer.Option('', '--uid', help='公司唯一标识uid（不传则使用当前已切换在用的 uid）'),
+):
+    service = TodoService()
+    res = service.set_month_audited(year, month, uid)
+    echo_util.echo_http_res(res)