Tonus 1.1.1__tar.gz

tonus-1.1.1/PKG-INFO

@@ -0,0 +1,135 @@
+Metadata-Version: 2.4
+Name: Tonus
+Version: 1.1.1
+Summary: a energy bar
+Author-email: firefly <your.email@example.com>
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: typer>=0.9.0
+Requires-Dist: rich
+Requires-Dist: typing_extensions
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+
+# Mana-bar (法力槽)
+
+Mana-bar 是一个基于命令行的任务流管理工具，旨在帮助你管理不同项目的“精力”或“疲劳值”。通过设定每个项目的疲劳上限和消耗速率，你可以更好地规划和追踪你的工作状态，避免过度劳累。
+
+## 简介
+
+在日常工作中，我们往往会在多个项目之间切换。Mana-bar 引入了“法力值”（Mana）或“疲劳值”的概念：
+- 每个项目都有一个**疲劳上限**（Total Fatigue）。
+- 当你开始在一个项目上工作时，疲劳值会随着时间积累。
+- 不同的项目可以设置不同的**消耗速率**（Rate）。
+- 当疲劳值达到上限时，意味着你需要休息或切换到其他项目。
+- 疲劳值会随着时间的推移（休息）通过某种机制恢复（当前代码逻辑主要关注记录和显示）。
+
+## 安装
+
+确保你已经安装了 Python 3.10+。
+
+```bash
+git clone <repository_url>
+cd mana_bar
+pip install .
+```
+
+或者使用 `pip` 安装依赖：
+```bash
+pip install -e .
+```
+
+## 使用说明
+
+安装完成后，你可以使用 `mana` 命令来管理你的任务。
+
+### 1. 添加项目 (Add)
+
+开始管理一个新项目之前，你需要先添加它，并设置其疲劳上限和消耗速率。
+
+```bash
+mana add <project_name> --total <total_fatigue> --rate <fatigue_rate>
+```
+
+- `<project_name>`: 项目名称。
+- `--total`, `-t`: 疲劳值上限，默认 156。
+- `--rate`, `-r`: 疲劳消耗速率，默认 1.0。
+
+示例：
+```bash
+mana add coding -t 200 -r 1.5
+```
+
+### 2. 开始任务 (Open)
+
+开始在一个项目上工作。
+
+```bash
+mana open <project_name> [--task <task_desc>] [--extra <extra_id>]
+```
+
+- `<project_name>`: 项目名称。
+- `--task`, `-t`: (可选) 任务描述。
+- `--extra`, `-e`: (可选) 额外ID。
+
+示例：
+```bash
+mana open coding -t "Implement README"
+```
+
+### 3. 结束任务 (Close)
+
+停止当前项目的计时。
+
+```bash
+mana close <project_name>
+```
+
+示例：
+```bash
+mana close coding
+```
+
+### 4. 查看项目状态 (Check)
+
+查看指定项目的当前疲劳值状态。
+
+```bash
+mana check <project_name>
+```
+
+输出将显示一个进度条，指示当前的精力和疲劳状态。
+
+### 5. 查看所有项目 (Checkall)
+
+查看所有项目的状态。
+
+```bash
+mana checkall
+```
+
+## 开发
+
+本项目使用 [Typer](https://typer.tiangolo.com/) 构建 CLI，使用 [Rich](https://github.com/Textualize/rich) 进行终端美化。
+
+### 依赖安装
+```bash
+pip install -r requirements.txt
+# 或者
+pip install .[dev]
+```
+
+### 运行测试
+```bash
+pytest
+```
+
+## License
+
+[MIT License](LICENSE)

tonus-1.1.1/README.md

@@ -0,0 +1,116 @@
+# Mana-bar (法力槽)
+
+Mana-bar 是一个基于命令行的任务流管理工具，旨在帮助你管理不同项目的“精力”或“疲劳值”。通过设定每个项目的疲劳上限和消耗速率，你可以更好地规划和追踪你的工作状态，避免过度劳累。
+
+## 简介
+
+在日常工作中，我们往往会在多个项目之间切换。Mana-bar 引入了“法力值”（Mana）或“疲劳值”的概念：
+- 每个项目都有一个**疲劳上限**（Total Fatigue）。
+- 当你开始在一个项目上工作时，疲劳值会随着时间积累。
+- 不同的项目可以设置不同的**消耗速率**（Rate）。
+- 当疲劳值达到上限时，意味着你需要休息或切换到其他项目。
+- 疲劳值会随着时间的推移（休息）通过某种机制恢复（当前代码逻辑主要关注记录和显示）。
+
+## 安装
+
+确保你已经安装了 Python 3.10+。
+
+```bash
+git clone <repository_url>
+cd mana_bar
+pip install .
+```
+
+或者使用 `pip` 安装依赖：
+```bash
+pip install -e .
+```
+
+## 使用说明
+
+安装完成后，你可以使用 `mana` 命令来管理你的任务。
+
+### 1. 添加项目 (Add)
+
+开始管理一个新项目之前，你需要先添加它，并设置其疲劳上限和消耗速率。
+
+```bash
+mana add <project_name> --total <total_fatigue> --rate <fatigue_rate>
+```
+
+- `<project_name>`: 项目名称。
+- `--total`, `-t`: 疲劳值上限，默认 156。
+- `--rate`, `-r`: 疲劳消耗速率，默认 1.0。
+
+示例：
+```bash
+mana add coding -t 200 -r 1.5
+```
+
+### 2. 开始任务 (Open)
+
+开始在一个项目上工作。
+
+```bash
+mana open <project_name> [--task <task_desc>] [--extra <extra_id>]
+```
+
+- `<project_name>`: 项目名称。
+- `--task`, `-t`: (可选) 任务描述。
+- `--extra`, `-e`: (可选) 额外ID。
+
+示例：
+```bash
+mana open coding -t "Implement README"
+```
+
+### 3. 结束任务 (Close)
+
+停止当前项目的计时。
+
+```bash
+mana close <project_name>
+```
+
+示例：
+```bash
+mana close coding
+```
+
+### 4. 查看项目状态 (Check)
+
+查看指定项目的当前疲劳值状态。
+
+```bash
+mana check <project_name>
+```
+
+输出将显示一个进度条，指示当前的精力和疲劳状态。
+
+### 5. 查看所有项目 (Checkall)
+
+查看所有项目的状态。
+
+```bash
+mana checkall
+```
+
+## 开发
+
+本项目使用 [Typer](https://typer.tiangolo.com/) 构建 CLI，使用 [Rich](https://github.com/Textualize/rich) 进行终端美化。
+
+### 依赖安装
+```bash
+pip install -r requirements.txt
+# 或者
+pip install .[dev]
+```
+
+### 运行测试
+```bash
+pytest
+```
+
+## License
+
+[MIT License](LICENSE)

tonus-1.1.1/Tonus.egg-info/PKG-INFO

@@ -0,0 +1,135 @@
+Metadata-Version: 2.4
+Name: Tonus
+Version: 1.1.1
+Summary: a energy bar
+Author-email: firefly <your.email@example.com>
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: typer>=0.9.0
+Requires-Dist: rich
+Requires-Dist: typing_extensions
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+
+# Mana-bar (法力槽)
+
+Mana-bar 是一个基于命令行的任务流管理工具，旨在帮助你管理不同项目的“精力”或“疲劳值”。通过设定每个项目的疲劳上限和消耗速率，你可以更好地规划和追踪你的工作状态，避免过度劳累。
+
+## 简介
+
+在日常工作中，我们往往会在多个项目之间切换。Mana-bar 引入了“法力值”（Mana）或“疲劳值”的概念：
+- 每个项目都有一个**疲劳上限**（Total Fatigue）。
+- 当你开始在一个项目上工作时，疲劳值会随着时间积累。
+- 不同的项目可以设置不同的**消耗速率**（Rate）。
+- 当疲劳值达到上限时，意味着你需要休息或切换到其他项目。
+- 疲劳值会随着时间的推移（休息）通过某种机制恢复（当前代码逻辑主要关注记录和显示）。
+
+## 安装
+
+确保你已经安装了 Python 3.10+。
+
+```bash
+git clone <repository_url>
+cd mana_bar
+pip install .
+```
+
+或者使用 `pip` 安装依赖：
+```bash
+pip install -e .
+```
+
+## 使用说明
+
+安装完成后，你可以使用 `mana` 命令来管理你的任务。
+
+### 1. 添加项目 (Add)
+
+开始管理一个新项目之前，你需要先添加它，并设置其疲劳上限和消耗速率。
+
+```bash
+mana add <project_name> --total <total_fatigue> --rate <fatigue_rate>
+```
+
+- `<project_name>`: 项目名称。
+- `--total`, `-t`: 疲劳值上限，默认 156。
+- `--rate`, `-r`: 疲劳消耗速率，默认 1.0。
+
+示例：
+```bash
+mana add coding -t 200 -r 1.5
+```
+
+### 2. 开始任务 (Open)
+
+开始在一个项目上工作。
+
+```bash
+mana open <project_name> [--task <task_desc>] [--extra <extra_id>]
+```
+
+- `<project_name>`: 项目名称。
+- `--task`, `-t`: (可选) 任务描述。
+- `--extra`, `-e`: (可选) 额外ID。
+
+示例：
+```bash
+mana open coding -t "Implement README"
+```
+
+### 3. 结束任务 (Close)
+
+停止当前项目的计时。
+
+```bash
+mana close <project_name>
+```
+
+示例：
+```bash
+mana close coding
+```
+
+### 4. 查看项目状态 (Check)
+
+查看指定项目的当前疲劳值状态。
+
+```bash
+mana check <project_name>
+```
+
+输出将显示一个进度条，指示当前的精力和疲劳状态。
+
+### 5. 查看所有项目 (Checkall)
+
+查看所有项目的状态。
+
+```bash
+mana checkall
+```
+
+## 开发
+
+本项目使用 [Typer](https://typer.tiangolo.com/) 构建 CLI，使用 [Rich](https://github.com/Textualize/rich) 进行终端美化。
+
+### 依赖安装
+```bash
+pip install -r requirements.txt
+# 或者
+pip install .[dev]
+```
+
+### 运行测试
+```bash
+pytest
+```
+
+## License
+
+[MIT License](LICENSE)

tonus-1.1.1/Tonus.egg-info/SOURCES.txt

@@ -0,0 +1,30 @@
+README.md
+pyproject.toml
+Tonus.egg-info/PKG-INFO
+Tonus.egg-info/SOURCES.txt
+Tonus.egg-info/dependency_links.txt
+Tonus.egg-info/entry_points.txt
+Tonus.egg-info/requires.txt
+Tonus.egg-info/top_level.txt
+package/__init__.py
+package/cli/__init__.py
+package/cli/app.py
+package/cli/main.py
+package/data/__init__.py
+package/data/init.py
+package/data/project.py
+package/data/record.py
+package/entity/__init__.py
+package/entity/project.py
+package/entity/record.py
+package/service/__init__.py
+package/service/project/__init__.py
+package/service/project/add.py
+package/service/project/delete.py
+package/service/project/extra.py
+package/service/record/__init__.py
+package/service/record/complete.py
+package/service/record/show.py
+package/service/record/start.py
+package/service/record/stop.py
+package/utils/utils.py

tonus-1.1.1/Tonus.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

tonus-1.1.1/Tonus.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+to = package.cli.main:main

tonus-1.1.1/Tonus.egg-info/requires.txt

@@ -0,0 +1,6 @@
+typer>=0.9.0
+rich
+typing_extensions
+
+[dev]
+pytest>=7.0.0

tonus-1.1.1/Tonus.egg-info/top_level.txt

@@ -0,0 +1 @@
+package

tonus-1.1.1/package/cli/app.py

@@ -0,0 +1,79 @@
+import typer
+from package.service.record.start import start
+from package.service.record.stop import stop
+from package.service.record.show import show, list_all
+from package.service.record.complete import complete_project_name
+from package.service.project.add import add_project
+from package.service.project.delete import delete_project
+from package.service.project.extra import extra_time
+
+from typing_extensions import Annotated
+
+app = typer.Typer(help="to")
+
+
+@app.callback(invoke_without_command=True)
+def main(ctx: typer.Context):
+    if ctx.invoked_subcommand is None:
+        list_all()
+
+
+@app.command()
+def open(
+    project: Annotated[
+        str, typer.Argument(help="project", shell_complete=complete_project_name)
+    ],
+    task_desc: Annotated[str, typer.Option("--task", "-t", help="task")] = "",
+    extra_id: Annotated[str, typer.Option("--extra", "-e", help="extra_id")] = "",
+):
+    start(project_name=project, task_desc=task_desc, extra_id=extra_id)
+
+
+@app.command()
+def close(
+    project: Annotated[
+        str, typer.Argument(help="project", shell_complete=complete_project_name)
+    ],
+):
+    stop(project_name=project)
+
+
+@app.command()
+def check(
+    project: Annotated[
+        str, typer.Argument(help="project", shell_complete=complete_project_name)
+    ],
+):
+    show(project)
+
+
+@app.command()
+def checkall():
+    list_all()
+
+
+@app.command()
+def add(
+    project: Annotated[str, typer.Argument(help="project")],
+    total: Annotated[int, typer.Option("--total", "-t", help="task")] = 156,
+    rate: Annotated[float, typer.Option("--extra", "-r", help="extra_id")] = 1.0,
+):
+    add_project(project_name=project, total=total, rate=rate)
+
+
+@app.command()
+def delete(
+    project: Annotated[
+        str, typer.Argument(help="project", shell_complete=complete_project_name)
+    ],
+):
+    delete_project(project_name=project)
+
+
+@app.command()
+def extra(
+    project: Annotated[str, typer.Argument(help="project")],
+    extra_num: Annotated[int,typer.Argument(help="extra minute")] = 0,
+    invalid_day: Annotated[int,typer.Argument(help="invalid day")] = 1,
+):
+    extra_time(project_name=project, minutes=extra_num, invalid_day=invalid_day)

tonus-1.1.1/package/cli/main.py

@@ -0,0 +1,13 @@
+from package.cli.app import app
+import typer
+
+
+def main():
+    try:
+        app()
+    except Exception as e:
+        typer.secho(f"出现异常: {e}", fg=typer.colors.RED, err=True)
+
+
+if __name__ == "__main__":
+    main()

tonus-1.1.1/package/data/init.py

@@ -0,0 +1,76 @@
+import sqlite3
+import functools
+import os
+from pathlib import Path
+import json
+
+
+_is_initialized = False
+
+
+DB_DIR = str(Path.home() / ".config" / "mana" / "data")
+DB_PATH = os.path.join(DB_DIR, "fatigue.db")
+CONFIG_PATH = os.path.join(DB_DIR, "config.json")
+
+
+def _init_db():
+    global _is_initialized
+
+    if _is_initialized:
+        return
+    conn = get_conn()
+    cursor = conn.cursor()
+    cursor.executescript(
+        """
+        CREATE TABLE IF NOT EXISTS records (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            start INTEGER,
+            stop INTEGER,
+            duration INTEGER,   
+            task_desc TEXT,
+            extra_id TEXT,
+            project_id INTEGER
+        );
+        
+        CREATE TABLE IF NOT EXISTS projects (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            project_name TEXT UNIQUE,
+            fatigue_rate REAL DEFAULT 1.0,
+            fatigue_total INTEGER DEFAULT 156,
+            project_status INTEGER DEFAULT 0,
+            extra_num INTEGER
+        );
+    """
+    )
+
+    conn.commit()
+    conn.close()
+    _is_initialized = True
+
+
+def _create_config():
+    if os.path.exists(CONFIG_PATH):
+        return
+    else:
+        with open(CONFIG_PATH, "w", encoding="utf-8") as f:
+            default_config = {"enable_timew": True}
+            json.dump(default_config, f, indent=4)
+
+
+def get_conn():
+    if not os.path.exists(DB_DIR):
+        os.makedirs(DB_DIR, exist_ok=True)
+
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+
+def ensure_db_ready(func):
+    @functools.wraps(func)
+    def wrapper(*args, **kargs):
+        _init_db()
+        _create_config()
+        return func(*args, **kargs)
+
+    return wrapper

tonus-1.1.1/package/data/project.py

@@ -0,0 +1,105 @@
+from package.data.init import ensure_db_ready, get_conn
+from package.entity.project import Project
+
+
+@ensure_db_ready
+def add_project(project: Project):
+    sql = "insert into projects(project_name,project_status,fatigue_rate,fatigue_total) values(?,?,?,?)"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (
+                project.project_name,
+                project.project_status,
+                project.fatigue_rate,
+                project.fatigue_total,
+            ),
+        )
+
+    return cursor.lastrowid
+
+@ensure_db_ready
+def delete_project(project: str):
+    sql = "delete from projects where project_name = ?"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (project,),
+        )
+
+
+@ensure_db_ready
+def update_project(project: Project):
+    project_dict = vars(project)
+    update_parts = {k: v for k,v in project_dict.items() if v is not None and k != "id"}
+    if not update_parts:
+        return None
+    update_keys = ",".join([f"{k} = ?" for k in update_parts.keys()])
+    sql = "update projects set " + update_keys + " where id = ?"
+    params = list(update_parts.values())
+    params.append(project.id)
+
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(sql, params)
+        conn.commit()
+
+@ensure_db_ready
+def batch_init_extra(project_ids: list[int]):
+    sql = "UPDATE projects SET extra_time = 0, invalid_time = 0, extra_start_time = 0 WHERE id = ?"
+    params = [(pid,) for pid in project_ids]
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.executemany(sql, params)
+        conn.commit()
+
+
+@ensure_db_ready
+def get_project_id(id: int):
+    sql = "select id,project_name,fatigue_rate,fatigue_total,project_status from projects where id = ?"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (id,),
+        )
+
+        return Project(**dict(cursor.fetchone()))
+
+
+@ensure_db_ready
+def get_project_name(project_name):
+    sql = "select id,project_name,fatigue_total,fatigue_rate,project_status,extra_time,extra_start_time,invalid_time from projects where project_name = ?"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (project_name,),
+        )
+        row = cursor.fetchone()
+        if not row:
+            return None
+
+        return Project(**dict(row))
+
+
+@ensure_db_ready
+def get_all_projects():
+    sql = (
+        "select id,project_name,fatigue_total,fatigue_rate,project_status,invalid_time,extra_time,extra_start_time from projects"
+    )
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+        )
+        rows = cursor.fetchall()
+        if not rows:
+            return None
+
+        results = []
+        for r in rows:
+            results.append(Project(**dict(r)))
+        return results

tonus-1.1.1/package/data/record.py

@@ -0,0 +1,162 @@
+from package.data.init import ensure_db_ready, get_conn
+from package.entity.record import Record
+from package.entity.project import Project
+from datetime import datetime, timedelta
+import time
+
+
+@ensure_db_ready
+def add_record(record: Record):
+    sql = "insert into records(start,stop,duration,task_desc,extra_id,project_id) values(?,?,?,?,?,?)"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (
+                record.start,
+                record.stop,
+                record.duration,
+                record.task_desc,
+                record.extra_id,
+                record.project_id,
+            ),
+        )
+        conn.commit()
+
+
+@ensure_db_ready
+def update_stop_record(record: Record):
+    sql = (
+        "update records set stop = ? , duration = ? where project_id = ? and stop = -1"
+    )
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (
+                record.stop,
+                record.duration,
+                record.project_id,
+            ),
+        )
+        conn.commit()
+
+
+@ensure_db_ready
+def get_recent_pending_record_by_project(record: Record):
+    sql = (
+        "select id,start,stop,task_desc from records where stop = -1 and project_id = ?"
+    )
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (record.project_id,),
+        )
+        row = cursor.fetchone()
+
+        if not row:
+            return None
+
+        return Record(**dict(row))
+
+
+@ensure_db_ready
+def get_recent_record_by_project(record: Record):
+    sql = "select id,start,stop,task_desc from records where project_id = ? ORDER BY start DESC LIMIT 1"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            (record.project_id,),
+        )
+        row = cursor.fetchone()
+
+        if not row:
+            return None
+
+        return Record(**dict(row))
+
+
+@ensure_db_ready
+def get_projects():
+    sql = "select * from projects"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql
+        )
+        rows = cursor.fetchall()
+
+        if not rows:
+            return None
+        return [Project(**dict(row)) for row in rows]
+
+
+@ensure_db_ready
+def get_today_accumulated_fatigue(project_id=None):
+    now = datetime.now()
+    # 截止今天0点的时间戳
+    # 如果现在还没到凌晨 4 点，说明还在“昨天的那个统计周期”里
+    if now.hour < 4:
+        start_time = now.replace(hour=4, minute=0, second=0, microsecond=0) - timedelta(
+            days=1
+        )
+    else:
+        # 否则，从今天的 4 点开始算
+        start_time = now.replace(hour=4, minute=0, second=0, microsecond=0)
+
+    start_ts = int(start_time.timestamp())
+    now_ts = int(time.time())
+
+    sql = """
+          SELECT p.id,
+                 p.project_name,
+                 p.fatigue_total,
+                 SUM(
+                         CASE
+                             WHEN r.stop = -1 then (? - start)
+                             ELSE r.duration
+                             END * p.fatigue_rate / 60
+                 ) AS used_fatigue,
+                 p.project_status,
+                 p.extra_time,
+                 p.invalid_time
+          FROM projects p
+                   LEFT JOIN records r ON r.project_id = p.id AND r.start >= ? \
+ \
+          """
+    params = [now_ts, start_ts]
+    if project_id:
+        sql = sql + " WHERE project_id = ?"
+        params.append(project_id)
+
+    sql += " GROUP BY p.id"
+    with get_conn() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            sql,
+            tuple(params),
+        )
+
+        if project_id:
+            row = cursor.fetchone()
+            if row is not None:
+                total_duration = row[3] if row[3] is not None else 0
+                return total_duration
+            else:
+                return 0
+        else:
+            rows = cursor.fetchall()
+            results = []
+            for r in rows:
+                results.append(
+                    {
+                        "id": r[0],
+                        "name": r[1],
+                        "total": r[2] + r[5],
+                        "used": r[3] if r[3] is not None else 0,
+                        "status": r[4],
+                    }
+                )
+            return results

tonus-1.1.1/package/entity/project.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class Project:
+    id: Optional[int]
+    project_name: Optional[str] = ""
+    fatigue_rate: Optional[float] = 1.0
+    fatigue_total: Optional[int] = 156
+    project_status: Optional[int] = 0,
+    extra_time: Optional[int] = 0,
+    invalid_time: Optional[int] = 0,
+    extra_start_time: Optional[int] = 0,

tonus-1.1.1/package/entity/record.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class Record:
+    id: Optional[int]
+    start: int = -1
+    stop: int = -1
+    duration: int = -1
+    task_desc: str = ""
+    extra_id: str = ""
+    project_id: int = -1

tonus-1.1.1/package/service/project/add.py

@@ -0,0 +1,15 @@
+from package.entity.project import Project
+from package.data.project import add_project as add_project_data
+from package.data.init import ensure_db_ready
+
+
+@ensure_db_ready
+def add_project(project_name, total, rate):
+    if project_name == "" or project_name is None:
+        raise ValueError("Project name cannot be empty")
+
+    project = Project(
+        None, project_name=project_name, fatigue_rate=rate, fatigue_total=total,project_status=3
+    )
+
+    add_project_data(project)

tonus-1.1.1/package/service/project/delete.py

@@ -0,0 +1,7 @@
+from package.data.project import delete_project as delete_project_data
+from package.data.init import ensure_db_ready
+
+
+@ensure_db_ready
+def delete_project(project_name):
+    delete_project_data(project_name)

tonus-1.1.1/package/service/project/extra.py

@@ -0,0 +1,17 @@
+from datetime import datetime
+
+from package.data.project import update_project, get_project_name
+from package.data.init import ensure_db_ready
+
+
+@ensure_db_ready
+def extra_time(project_name, minutes:int, invalid_day:int):
+    invalid_time = invalid_day * 24 * 60 * 60 * 60
+    extra_start_time = int(datetime.now().timestamp())
+    if project_name == "" or project_name is None:
+        raise ValueError("Project name cannot be empty")
+    project = get_project_name(project_name=project_name)
+    project.extra_time = project.extra_time + minutes
+    project.invalid_time = invalid_time
+    project.extra_start_time = extra_start_time
+    update_project(project)

tonus-1.1.1/package/service/record/complete.py

@@ -0,0 +1,14 @@
+from package.data.project import get_all_projects
+import typer
+
+
+def complete_project_name(ctx: typer.Context, param, incomplete: str):
+    text = incomplete
+    projects = get_all_projects()
+    choices = [
+        p.project_name
+        for p in projects
+        if isinstance(p.project_name, str) and p.project_name.startswith(text)
+    ]
+
+    return choices

tonus-1.1.1/package/service/record/show.py

@@ -0,0 +1,105 @@
+from datetime import datetime
+from package.data.project import get_project_name,get_all_projects,batch_init_extra
+from package.data.record import (
+    get_today_accumulated_fatigue,
+)
+from rich.console import Console
+from rich.table import Table
+from rich.progress_bar import ProgressBar
+from rich.text import Text
+import typer
+
+status_map = {
+    0 : ("已暂停", "red"),
+    1 : ("进行中", "green"),
+    2 : ("已结束", "grey"),
+    3 : ("未开始", "yellow")
+}
+
+def show(project_name):
+    console = Console()
+    project = get_project_name(project_name=project_name)
+
+    if not project:
+        typer.secho("找不到此项目", fg=typer.colors.RED, err=True)
+        return
+
+    cost = int(get_today_accumulated_fatigue(project.id))
+
+    if cost > project.fatigue_total:
+        cost = project.fatigue_total
+
+    rest = (project.fatigue_total + project.extra_time) - cost
+    total = project.fatigue_total + project.extra_time
+
+    if rest < 0:
+        rest = 0
+
+    bar = ProgressBar(
+        total=total,
+        completed=rest,
+        width=40,
+        complete_style="yellow",
+        finished_style="green",
+    )
+    status_text, color = status_map.get(project.project_status,("未知","white"))
+    status = Text(status_text, style=color)
+    grid = Table.grid(padding=1)
+    grid.add_row(
+        Text(
+            f"{project.project_name}: {rest}/{total}:",
+            style="bold blue",
+        ),
+        status,
+        bar,
+        Text(f"{rest / total:.1%}", style="magenta"),
+    )
+    console.print(grid)
+
+
+def list_all():
+    p_ids = confirm_extra()
+    batch_init_extra(p_ids)
+    results = get_today_accumulated_fatigue()
+    console = Console()
+    if not results:
+        typer.secho("没有记录，请确认", fg=typer.colors.RED, err=True)
+    grid = Table.grid(padding=1)
+    grid.add_column(justify="right")
+    grid.add_column(justify="left")
+    grid.add_column(justify="right")
+    grid = Table.grid(expand=True, padding=(0, 1))
+    grid.add_column(no_wrap=True, justify="left")
+    grid.add_column(no_wrap=True, justify="right")
+    grid.add_column(no_wrap=True, justify="center", width=10)
+    grid.add_column(ratio=1)
+    grid.add_column(no_wrap=True, justify="right")
+
+    for r in results:
+        rest = max(0, int(r["total"] - r["used"]))
+        bar = ProgressBar(
+            total=r["total"],
+            completed=rest,
+            width=None, # 宽度自适应
+            complete_style="yellow",
+            finished_style="green",
+        )
+
+        status_text, color = status_map.get(r["status"], ("未知", "white"))
+        status = Text(status_text, style=color)
+        grid.add_row(
+            Text(f"{r['id']} {r['name']}:", style="bold blue"),
+            Text(f"{rest}/{r['total']}:", style="bold blue"),
+            status,
+            bar,
+            Text(f"{rest / r['total']:.1%}", style="magenta")
+        )
+    console.print(grid)
+
+
+def confirm_extra():
+    projects = get_all_projects()
+    current_stamp = datetime.now().timestamp()
+    filtered_projects = filter(lambda p: int(current_stamp) >= p.invalid_time + p.extra_start_time, projects)
+    project_ids = list(map(lambda p: p.id, filtered_projects))
+    return project_ids

tonus-1.1.1/package/service/record/start.py

@@ -0,0 +1,34 @@
+from package.data.project import get_project_name, add_project, update_project
+from package.data.record import add_record, get_recent_pending_record_by_project
+from package.entity.project import Project
+from package.entity.record import Record
+from package.utils.utils import get_config
+from datetime import datetime
+import typer
+import subprocess
+
+
+def start(task_desc, project_name, extra_id):
+    p = get_project_name(project_name)
+    if not p:
+        p = Project(None, project_name=project_name, project_status=1)
+        p_id = add_project(p)
+    else:
+        p_id = p.id
+
+    record = get_recent_pending_record_by_project(Record(id=None, project_id=p_id))
+    if record:
+        typer.secho("项目已经进行，请先停止", fg=typer.colors.RED, err=True)
+        return
+    start_time = int(datetime.today().timestamp())
+    record = Record(
+        None, start=start_time, task_desc=task_desc, project_id=p_id, extra_id=extra_id
+    )
+    add_record(record=record)
+    p.project_status = 1
+    update_project(project=p)
+
+    config = get_config()
+    if config["enable_timew"]:
+        params = ["timew", "start", project_name]
+        subprocess.run(params)

tonus-1.1.1/package/service/record/stop.py

@@ -0,0 +1,33 @@
+from package.data.project import get_project_name, update_project
+from package.data.record import get_recent_pending_record_by_project, update_stop_record
+from package.entity.record import Record
+from package.utils.utils import get_config
+from datetime import datetime
+import typer
+import subprocess
+
+
+def stop(project_name):
+    project = get_project_name(project_name=project_name)
+    total_second = project.fatigue_total * 60
+    used_total_second = project.fatigue_total * total_second
+    if not project:
+        raise ValueError("找不到项目，请重新输入")
+    if not (record_ := get_recent_pending_record_by_project(Record(id=None, project_id=project.id))):
+        typer.secho("找不到正在进行的记录")
+        return
+    start_time = record_.start
+    if not start_time:
+        raise ValueError("找不到开始时间，请重新输入")
+    stop_time = int(datetime.today().timestamp())
+    duration_time = stop_time - start_time
+    record = Record(
+        id=None, stop=stop_time, duration=duration_time, project_id=project.id
+    )
+    update_stop_record(record)
+    project.project_status = 0 if used_total_second >= duration_time else 1
+    update_project(project=project)
+    config = get_config()
+    if config["enable_timew"]:
+        params = ["timew", "stop", project_name]
+        subprocess.run(params)

tonus-1.1.1/package/utils/utils.py

@@ -0,0 +1,11 @@
+from pathlib import Path
+import os
+import json
+
+CONFIG_DIR = str(Path.home() / ".config" / "mana" / "data")
+CONFIG_PATH = os.path.join(CONFIG_DIR, "config.json")
+
+
+def get_config():
+    with open(CONFIG_PATH, "r", encoding="utf-8") as f:
+        return json.load(f)

tonus-1.1.1/pyproject.toml

@@ -0,0 +1,40 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "Tonus"
+version = "1.1.1"
+description = "a energy bar"
+readme = "README.md"
+requires-python = ">=3.10"
+authors = [
+    {name = "firefly", email = "your.email@example.com"}
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+
+dependencies = [
+    "typer>=0.9.0",
+    "rich",
+    "typing_extensions"
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0.0",
+]
+
+[project.scripts]
+to = "package.cli.main:main"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["package*"]
+exclude = ["test*"]

tonus-1.1.1/setup.cfg

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