e.vlasov/redmine_tracker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Добавил новые таблицы

Browse Source
This commit is contained in:
quicklike
2025-05-20 23:13:03 +03:00
parent 39ede4f880
commit 9ca773681c
2 changed files with 74 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

117
db/db.py
View File

@@ -2,55 +2,64 @@ import logging
import sqlite3 as sq
from classes import Activity, Tag
from enums import ActivityType
class Database:
def __init__(self, db_name: str):
try:
self.con = sq.connect(db_name)
self.con.cursor().executescript('''CREATE TABLE IF NOT EXISTS "activities" (
"id" VARCHAR NOT NULL UNIQUE,
"description" VARCHAR NOT NULL,
"time_spent" NUMERIC NOT NULL,
"date_start" DATETIME NOT NULL,
"date_end" DATETIME,
"is_tracked" BOOLEAN NOT NULL DEFAULT false,
"vetro_project_id" VARCHAR NOT NULL,
"tag_id" VARCHAR NOT NULL,
"project_id" VARCHAR NOT NULL,
PRIMARY KEY("id"),
FOREIGN KEY ("tag_id") REFERENCES "tags"("id")
FOREIGN KEY ("vetro_project_id") REFERENCES "vetro_projects"("id")
ON UPDATE NO ACTION ON DELETE NO ACTION
);
self.con.cursor().executescript('''
CREATE TABLE IF NOT EXISTS "activities" (
"id" VARCHAR NOT NULL UNIQUE,
"description" VARCHAR NOT NULL,
"time_spent" NUMERIC(3, 3) NOT NULL,
"date_start" DATETIME NOT NULL,
"date_end" DATETIME,
"is_tracked" BOOLEAN NOT NULL DEFAULT false,
"vetro_project_id" VARCHAR NOT NULL,
"tag_id" VARCHAR NOT NULL,
"project_id" VARCHAR NOT NULL,
PRIMARY KEY("id"),
FOREIGN KEY ("tag_id") REFERENCES "tags"("id")
FOREIGN KEY ("vetro_project_id") REFERENCES "vetro_projects"("id")
ON UPDATE NO ACTION ON DELETE NO ACTION
);
CREATE TABLE IF NOT EXISTS "tags" (
"id" VARCHAR NOT NULL UNIQUE,
"title" VARCHAR NOT NULL,
PRIMARY KEY("id")
);
CREATE TABLE IF NOT EXISTS "tags" (
"id" VARCHAR NOT NULL UNIQUE,
"title" VARCHAR NOT NULL,
PRIMARY KEY("id")
);
CREATE TABLE IF NOT EXISTS "vetro_projects" (
"id" VARCHAR NOT NULL UNIQUE,
"title" VARCHAR NOT NULL,
PRIMARY KEY("id")
);
CREATE TABLE IF NOT EXISTS "vetro_projects" (
"id" VARCHAR NOT NULL UNIQUE,
"title" VARCHAR NOT NULL,
PRIMARY KEY("id")
);
CREATE TABLE IF NOT EXISTS "activity_types" (
"id" INTEGER NOT NULL UNIQUE,
"title" VARCHAR NOT NULL,
PRIMARY KEY("id")
);
CREATE TABLE IF NOT EXISTS "time_entry" (
"id" INTEGER NOT NULL UNIQUE,
"activity_id" VARCHAR NOT NULL,
"task_id" INTEGER NOT NULL,
"date" DATETIME NOT NULL,
"time_spent" NUMERIC NOT NULL,
"description" VARCHAR NOT NULL,
"activity_type" INTEGER NOT NULL CHECK(activity_type in (9, 8, 57, 58, 60)),
PRIMARY KEY("id"),
FOREIGN KEY ("activity_id") REFERENCES "activities"("id")
ON UPDATE NO ACTION ON DELETE NO ACTION,
FOREIGN KEY ("task_id") REFERENCES "activities"("task_id")
ON UPDATE NO ACTION ON DELETE NO ACTION
);
''')
CREATE TABLE IF NOT EXISTS "time_entry" (
"id" INTEGER PRIMARY KEY,
"date" DATETIME NOT NULL,
"description" VARCHAR NOT NULL,
"activity_id" VARCHAR NOT NULL,
"time_spent" NUMERIC(3, 3) NOT NULL,
"task_id" INTEGER NOT NULL,
"activity_type_id" INTEGER NOT NULL,
FOREIGN KEY ("activity_id") REFERENCES "activities"("id")
ON UPDATE NO ACTION ON DELETE NO ACTION,
FOREIGN KEY ("task_id") REFERENCES "activities"("task_id")
ON UPDATE NO ACTION ON DELETE NO ACTION,
FOREIGN KEY ("activity_type_id") REFERENCES "activity_types"("id")
ON UPDATE NO ACTION ON DELETE NO ACTION
);
''')
except Exception as e:
raise(e)
@@ -82,19 +91,16 @@ CREATE TABLE IF NOT EXISTS "time_entry" (
for act in activities:
try:
cur.execute(
"INSERT INTO time_entry VALUES(?, ?, ? ,? ,? ,? ,? ,? ,? ,? ,? ,?)",
"""INSERT INTO time_entry
(date, description, activity_id, time_spent, task_id, activity_type_id)
VALUES(?,?,?,?,?,?)""",
(
act.id,act.vetro_project_id,
act.vetro_project_title,
act.task_id,
act.date_start.date(),
act.description,
act.task_title,
act.tag_id,
act.project_id,
act.id,
act.time_spent,
act.date_start,
act.date_end,
act.is_tracked
act.task_id,
ActivityType.DEVELOPMENT
)
)
self.con.commit()
@@ -127,6 +133,11 @@ CREATE TABLE IF NOT EXISTS "time_entry" (
except Exception as e:
logging.error(e)
def insert_activity_types(self, activity_types: dict):
cur = self.con.cursor()
for k,v in activity_types.items():
try:
cur.execute("INSERT INTO activity_types VALUES(?, ?)", (k, v))
self.con.commit()
except Exception as e:
logging.error(e)