awork create Task

awork/awork_tasks.py

@@ -1,9 +1,14 @@
 from credentials import Client
+import typer
 import requests
 import json
 import datetime
 from pathlib import Path
+from model import AworkProjekt, Comment, User
+from sqlalchemy import create_engine, text
+from sqlalchemy.orm import Session
 
+DSN = "mysql+pymysql://gaps:Gcbs12ma@192.168.2.41/tasks"
 
 awork_api_url = "https://api.awork.com/api/v1"
 token_expires_in = 24 * 60 * 60
@@ -17,18 +22,139 @@ def main():
     bearer_token = access_token_file.read_text()
     header["Authorization"] = f"Bearer {bearer_token}"
 
-    users()
-    companies()
+    # users = get_users()
+    # companies = get_companies()
+    # import_projects()
+    # create_task(73849)
 
 
-def companies():
+def get_companies():
     res = requests.get(awork_api_url + "/companies", headers=header)
-    json.dump(res.json(), Path(__file__).parent.joinpath("companies.json").open("w"), indent=2)
+    data = res.json()
+    json.dump(data, Path(__file__).parent.joinpath("companies.json").open("w"), indent=2)
+    return data
 
 
-def users():
+def get_users():
     res = requests.get(awork_api_url + "/users", headers=header)
-    json.dump(res.json(), Path(__file__).parent.joinpath("users.json").open("w"), indent=2)
+    data = res.json()
+    json.dump(data, Path(__file__).parent.joinpath("users.json").open("w"), indent=2)
+    return data
+
+
+def import_projects():
+    projects = get_projects()
+    p = convert_projects(projects)
+
+    with Session(create_engine(DSN)) as db_session:
+        db_session.execute(text("TRUNCATE TABLE awork_projekte"))
+        db_session.add_all(p)
+        db_session.commit()
+
+
+def get_projects():
+    res = requests.get(awork_api_url + "/projects?pageSize=1000", headers=header)
+    data = res.json()
+    json.dump(data, Path(__file__).parent.joinpath("projects.json").open("w"), indent=2)
+    return data
+
+
+def convert_projects(projects):
+    res = []
+    for p in projects:
+        if "companyId" not in p:
+            continue
+
+        p_allgemein = 1 if "Allgemein" in p["name"] else 0
+        p_typ = p["projectType"]["name"] if "projectType" in p else "Unbekannt"
+
+        entry = AworkProjekt(
+            **{
+                "awork_project_id": p["id"],
+                "awork_company_id": p["companyId"],
+                "projekt_name": p["name"],
+                "projekt_status": p["projectStatus"]["name"],
+                "projekt_typ": p_typ,
+                "projekt_allgemein": p_allgemein,
+                "kunde_name": p["company"]["name"],
+                "tasks_count": p["tasksCount"],
+                "tracked_duration": p["trackedDuration"],
+            }
+        )
+        res.append(entry)
+    return res
+
+
+def create_task(comment_id: int):
+    with Session(create_engine(DSN)) as db_session:
+        comment = db_session.get(Comment, comment_id)
+
+        if comment.awork_task_id:
+            return comment.awork_task_id
+
+        datum = comment.datum.strftime("%d.%m.%Y")
+        link = f"http://gc-server1/fehlerbericht/#/report/{comment.kunde}/{comment.datum}/{comment.start}"
+
+        res = requests.get(awork_api_url + "/projects/" + comment.awork_project_id + "/taskstatuses", headers=header)
+        data = res.json()
+        task_status = [s["id"] for s in data if s["name"] == "To do"]
+        if len(task_status) == 0:
+            return 0
+        task_status_id = task_status[0]
+
+        task = {
+            "name": f"{comment.kunde} - Fehlerbericht vom {datum} - {comment.fehler} Fehler",
+            "description": (
+                f'<p><a target="_blank" rel="noopener noreferrer nofollow" href="{link}">Fehlerbericht-Portal</a></p>'
+                + f"<p>{comment.benutzer}: {comment.kommentar}</p>"
+            ),
+            "isPrio": False,
+            # "startOn": "2021-03-03T17:00:00Z",
+            "dueOn": f"{comment.datum}T16:00:00Z",
+            # "laneOrder": 0,
+            "plannedDuration": 3600,
+            # "remainingDuration": 0,
+            "typeOfWorkId": "1854242b-8d9a-44c6-a98e-f17b89b6bed8",
+            "taskStatusId": task_status_id,
+            # "order": 0,
+            # "subtaskOrder": 1,
+            "entityId": comment.awork_project_id,
+            "baseType": "projecttask",
+            # "parentId": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
+            # "lists": [{"id": "94166142-b5f9-4bd0-968d-a6a3474ee705", "order": 0}],
+        }
+        res = requests.post(awork_api_url + "/tasks", json=task, headers=header)
+        data = res.json()
+        task_id = data["id"]
+
+        res = requests.post(
+            awork_api_url + f"/tasks/{task_id}/addtags", json=[{"name": "Fehlerprotokolle"}], headers=header
+        )
+
+        user1 = db_session.get(User, comment.benutzer)
+
+        user_comment = {
+            "message": comment.kommentar,
+            "userId": user1.awork_user_id,
+        }
+
+        res = requests.post(awork_api_url + f"/tasks/{task_id}/comments", json=user_comment, headers=header)
+
+        res = requests.post(
+            awork_api_url + f"/tasks/{task_id}/setassignees", json=[user1.awork_user_id], headers=header
+        )
+
+        user2 = user1
+        if comment.benutzer2:
+            user2 = db_session.get(User, comment.benutzer2)
+
+            res = requests.post(
+                awork_api_url + f"/tasks/{task_id}/setassignees", json=[user2.awork_user_id], headers=header
+            )
+
+        comment.awork_task_id = task_id
+
+        db_session.commit()
 
 
 def login():
@@ -116,3 +242,4 @@ def authorize():
 
 if __name__ == "__main__":
     main()
+    typer.run(create_task)

awork/model.py

@@ -0,0 +1,47 @@
+from sqlalchemy import Integer, String, Date, Time
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+from sqlalchemy.dialects.mysql.base import TINYINT, TEXT
+from datetime import date, time
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class AworkProjekt(Base):
+    __tablename__ = "awork_projekte"
+    awork_project_id: Mapped[str] = mapped_column(String(50), primary_key=True)
+    awork_company_id: Mapped[str] = mapped_column(String(50))
+    projekt_name: Mapped[str] = mapped_column(String(100))
+    projekt_status: Mapped[str] = mapped_column(String(20))
+    projekt_typ: Mapped[str] = mapped_column(String(50))
+    projekt_allgemein: Mapped[int] = mapped_column(TINYINT, default=0)
+    kunde_name: Mapped[str] = mapped_column(String(100))
+    tasks_count: Mapped[int] = mapped_column(Integer, default=0)
+    tracked_duration: Mapped[int] = mapped_column(Integer, default=0)
+
+
+class Comment(Base):
+    __tablename__ = "status_kommentar"
+    id: Mapped[int] = mapped_column(Integer, primary_key=True)
+    datum: Mapped[date] = mapped_column(Date)
+    kunde: Mapped[str] = mapped_column(String(50))
+    start: Mapped[time] = mapped_column(Time)
+    benutzer: Mapped[str] = mapped_column(String(50))
+    benutzer2: Mapped[str] = mapped_column(String(50), nullable=True)
+    fehler: Mapped[int] = mapped_column(Integer)
+    status: Mapped[int] = mapped_column(Integer)
+    kommentar: Mapped[TEXT] = mapped_column(TEXT)
+    awork_project_id: Mapped[str] = mapped_column(String(50), nullable=True)
+    awork_task_id: Mapped[str] = mapped_column(String(50), nullable=True)
+
+
+class User(Base):
+    __tablename__ = "benutzer"
+    benutzer: Mapped[str] = mapped_column(String(50), primary_key=True)
+    name: Mapped[str] = mapped_column(String(50))
+    vorname: Mapped[str] = mapped_column(String(50))
+    email: Mapped[str] = mapped_column(String(100))
+    passwort: Mapped[str] = mapped_column(String(255))
+    aktiv: Mapped[int] = mapped_column(TINYINT, default=1)
+    awork_user_id: Mapped[str] = mapped_column(String(50), nullable=True)