nlttask 1.0.2__py3-none-any.whl

funtask/__init__.py

@@ -0,0 +1 @@
+from .models import Task

funtask/__main__.py

@@ -0,0 +1,4 @@
+from .web.app import start_web_server
+
+if __name__ == "__main__":
+    start_web_server()

funtask/models/__init__.py

@@ -0,0 +1 @@
+from .base import Task

funtask/models/base.py

@@ -0,0 +1,6 @@
+class Task:
+    def __init__(self, *args, **kwargs):
+        pass
+
+    def run(self, *args, **kwargs):
+        pass

funtask/models/task_model.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+from sqlalchemy import create_engine, Column, Integer, String, DateTime, Text
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
+Base = declarative_base()
+
+
+class TaskModel(Base):
+    __tablename__ = "tasks"
+
+    id = Column(Integer, primary_key=True)
+    task_dir = Column(String(255), nullable=False)
+    status = Column(
+        String(50), default="pending"
+    )  # pending, running, completed, failed
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+    description = Column(Text, nullable=True)
+    task_type = Column(String(50))  # slurm, cpp, etc.
+    output = Column(Text, nullable=True)
+
+
+# 创建数据库连接
+engine = create_engine("sqlite:///tasks.db", echo=True)
+Base.metadata.create_all(engine)
+Session = sessionmaker(bind=engine)

funtask/task/manager.py

@@ -0,0 +1,50 @@
+from datetime import datetime
+from typing import List, Optional
+from ..models.task_model import TaskModel, Session
+
+
+class TaskManager:
+    def __init__(self):
+        self.session = Session()
+
+    def create_task(
+        self, task_dir: str, task_type: str, description: Optional[str] = None
+    ) -> TaskModel:
+        task = TaskModel(
+            task_dir=task_dir,
+            task_type=task_type,
+            description=description,
+            status="pending",
+        )
+        self.session.add(task)
+        self.session.commit()
+        return task
+
+    def get_task(self, task_id: int) -> Optional[TaskModel]:
+        return self.session.query(TaskModel).filter(TaskModel.id == task_id).first()
+
+    def get_all_tasks(self) -> List[TaskModel]:
+        return self.session.query(TaskModel).order_by(TaskModel.created_at.desc()).all()
+
+    def update_task_status(
+        self, task_id: int, status: str, output: Optional[str] = None
+    ) -> Optional[TaskModel]:
+        task = self.get_task(task_id)
+        if task:
+            task.status = status
+            task.updated_at = datetime.now()
+            if output:
+                task.output = output
+            self.session.commit()
+        return task
+
+    def delete_task(self, task_id: int) -> bool:
+        task = self.get_task(task_id)
+        if task:
+            self.session.delete(task)
+            self.session.commit()
+            return True
+        return False
+
+    def __del__(self):
+        self.session.close()

funtask/task/submit.py

@@ -0,0 +1,84 @@
+"""
+#!/bin/bash
+task_root='/work/home/liuc12/workbench'
+timestamp=$(date +"%Y%m%d%H%M%S")
+task_dir="$task_root/task_$timestamp"
+
+mkdir "$task_dir"
+cp -r *.cpp *.h *.sh *.slurm *.f90 *.dat  $task_dir 2>/dev/null
+cd "$task_dir"
+pwd
+#ls -al
+sbatch config.slurm
+"""
+
+import os
+from datetime import datetime
+
+from funbuild.shell import run_shell
+from nltlog import getLogger
+
+from .manager import TaskManager
+
+logger = getLogger("funbuild")
+
+
+def submit_task():
+    task_dir = os.path.join(
+        os.environ["HOME"], "/workbench", datetime.now().strftime("%Y%m%d%H%M%S")
+    )
+    logger.info(f"任务主目录：{task_dir}")
+    os.makedirs(task_dir, exist_ok=True)
+
+    # 创建任务管理器实例
+    task_manager = TaskManager()
+    task_type = None
+    description = None
+
+    logger.info(f"step1: 复制文件到任务主目录：{task_dir}")
+    run_shell(f"cp -r *.cpp *.h *.sh *.slurm *.f90 *.dat {task_dir} 2>/dev/null")
+
+    try:
+        if os.path.exists("config.slurm"):
+            task_type = "slurm"
+            description = "SLURM cluster task"
+            logger.info("step2: 检测到config.slurm文件，提交任务")
+            # 创建任务记录
+            task = task_manager.create_task(task_dir, task_type, description)
+
+            try:
+                output = run_shell(f"cd {task_dir} && sbatch config.slurm")
+                task_manager.update_task_status(task.id, "running", output)
+            except Exception as e:
+                task_manager.update_task_status(task.id, "failed", str(e))
+                raise e
+
+        elif os.path.exists("main.cpp"):
+            task_type = "cpp"
+            description = "Local C++ compilation and execution"
+            # 创建任务记录
+            task = task_manager.create_task(task_dir, task_type, description)
+
+            try:
+                logger.info("step2: 检测到main.cpp文件，编译")
+                compile_output = run_shell(f"cd {task_dir} && g++ main.cpp -o task.app")
+                logger.info("step3: 编译完成，开始执行")
+                execution_output = run_shell(f"cd {task_dir} && ./task.app")
+                task_manager.update_task_status(
+                    task.id,
+                    "completed",
+                    f"Compilation: {compile_output}\nExecution: {execution_output}",
+                )
+            except Exception as e:
+                task_manager.update_task_status(task.id, "failed", str(e))
+                raise e
+
+        logger.info("任务提交完成")
+        return task_dir
+    except Exception as e:
+        logger.error(f"任务执行失败: {str(e)}")
+        raise e
+
+
+if __name__ == "__main__":
+    submit_task()

funtask/web/app.py

@@ -0,0 +1,90 @@
+from nicegui import ui
+from ..task.manager import TaskManager
+from ..task.submit import submit_task
+
+task_manager = TaskManager()
+
+
+def create_task_list():
+    with ui.card().classes("w-full"):
+        ui.label("任务列表").classes("text-h6")
+
+        # 创建任务表格
+        columns = [
+            {"name": "id", "label": "ID", "field": "id"},
+            {"name": "task_dir", "label": "任务目录", "field": "task_dir"},
+            {"name": "status", "label": "状态", "field": "status"},
+            {"name": "task_type", "label": "任务类型", "field": "task_type"},
+            {"name": "created_at", "label": "创建时间", "field": "created_at"},
+            {"name": "description", "label": "描述", "field": "description"},
+        ]
+
+        tasks = task_manager.get_all_tasks()
+        rows = [
+            {
+                "id": task.id,
+                "task_dir": task.task_dir,
+                "status": task.status,
+                "task_type": task.task_type,
+                "created_at": task.created_at.strftime("%Y-%m-%d %H:%M:%S"),
+                "description": task.description,
+            }
+            for task in tasks
+        ]
+
+        table = ui.table(columns=columns, rows=rows, row_key="id").classes("w-full")
+
+        async def refresh_table():
+            tasks = task_manager.get_all_tasks()
+            rows = [
+                {
+                    "id": task.id,
+                    "task_dir": task.task_dir,
+                    "status": task.status,
+                    "task_type": task.task_type,
+                    "created_at": task.created_at.strftime("%Y-%m-%d %H:%M:%S"),
+                    "description": task.description,
+                }
+                for task in tasks
+            ]
+            table.rows = rows
+
+        async def delete_task(task_id: int):
+            if task_manager.delete_task(task_id):
+                ui.notify(f"任务 {task_id} 已删除")
+                await refresh_table()
+            else:
+                ui.notify(f"删除任务 {task_id} 失败", type="negative")
+
+        async def view_task_output(task_id: int):
+            task = task_manager.get_task(task_id)
+            if task and task.output:
+                with ui.dialog() as dialog, ui.card():
+                    ui.label(f"任务 {task_id} 输出").classes("text-h6")
+                    ui.textarea(value=task.output, readonly=True).classes("w-full")
+                    ui.button("关闭", on_click=dialog.close)
+                dialog.open()
+            else:
+                ui.notify("没有可用的输出", type="warning")
+
+        # 添加操作按钮
+        with ui.row():
+            ui.button("刷新", on_click=refresh_table).classes("mr-2")
+            for row in rows:
+                with ui.row():
+                    ui.button(
+                        "查看输出", on_click=lambda r=row: view_task_output(r["id"])
+                    ).classes("mr-2")
+                    ui.button(
+                        "删除", on_click=lambda r=row: delete_task(r["id"]), color="red"
+                    )
+
+
+@ui.page("/")
+def main_page():
+    ui.label("任务管理系统").classes("text-h4 q-mb-md")
+    create_task_list()
+
+
+def start_web_server(host="0.0.0.0", port=8080):
+    ui.run(host=host, port=port, title="任务管理系统")

nlttask-1.0.2.dist-info/METADATA

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: nlttask
+Version: 1.0.2
+Summary: nlttask
+Author-email: 牛哥 <niuliangtao@qq.com>, farfarfun <farfarfun@qq.com>
+Maintainer-email: 牛哥 <niuliangtao@qq.com>, farfarfun <farfarfun@qq.com>
+Project-URL: Organization, https://github.com/farfarfun
+Project-URL: Repository, https://github.com/farfarfun/funtask
+Project-URL: Releases, https://github.com/farfarfun/funtask/releases
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: nltlog>=1.1.3
+Requires-Dist: sqlalchemy>=2.0.50
+
+# notetask

nlttask-1.0.2.dist-info/RECORD

@@ -0,0 +1,13 @@
+funtask/__init__.py,sha256=1An9ji9Yqij_SZdHEcVm8V_HRgpd_D9QbAro9bGklVw,25
+funtask/__main__.py,sha256=vPcdHTqX2krxMpQfa58Vei7_lzQcGbOZiSxr5Iv7_hY,89
+funtask/models/__init__.py,sha256=WanozeTJqTPVIpaXQzB5lvDI9XJLPoioyEKWB9RYwUQ,23
+funtask/models/base.py,sha256=1mPWWKfwU4upK1xUshwcSCFHecTWpbZ86-FJvOYDzd4,116
+funtask/models/task_model.py,sha256=Q4W30WfQo0eKorR88SFHskwkRgXBiGluqUcGlRXf7r4,911
+funtask/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+funtask/task/manager.py,sha256=KzO9vUiPBsMTTzfO_WrdWxU56pC3AffjhDi5t8JomYU,1498
+funtask/task/submit.py,sha256=rILZLceHf0t8THZbL0eXzfUxvgPexDy3u4OA4y9R6e4,2681
+funtask/web/app.py,sha256=onrKVp290rVNqs3_0vjGfb7tfn6NoUSX3IuTYYo7_nk,3323
+nlttask-1.0.2.dist-info/METADATA,sha256=_jt-VcvjA1zRt2RLZj8XgHUCvyDxz8VbAvEGZ5PPl5I,541
+nlttask-1.0.2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+nlttask-1.0.2.dist-info/top_level.txt,sha256=svcT_-fhz7u77X8wJYILLorz0fp2LjdYljx7QGa6ieg,8
+nlttask-1.0.2.dist-info/RECORD,,

nlttask-1.0.2.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

nlttask-1.0.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+funtask