jkuhl/corrlib
1
0
Fork 0
Code Issues 4 Pull requests Releases Packages Activity Actions

ensure db get before query

Browse source
This commit is contained in:
Justus Kuhlmann 2025-11-21 09:42:40 +01:00
commit 2940ee9055
Signed by: jkuhl
GPG key ID: 00ED992DD79B85A6
4 changed files with 18 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

18
corrlib/main.py
View file

@ -24,13 +24,15 @@ def create_project(path: str, uuid: str, owner: Union[str, None]=None, tags: Uni
code: str (optional)
The code that was used to create the measurements.
"""
conn = sqlite3.connect(path + "/backlogger.db")
db = path + "/backlogger.db"
dl.get(db, dataset=path)
conn = sqlite3.connect(db)
c = conn.cursor()
known_projects = c.execute("SELECT * FROM projects WHERE id=?", (uuid,))
if known_projects.fetchone():
raise ValueError("Project already imported, use update_project() instead.")
dl.unlock(path + "/backlogger.db", dataset=path)
dl.unlock(db, dataset=path)
alias_str = None
if aliases is not None:
alias_str = list2str(aliases)
@ -40,10 +42,11 @@ def create_project(path: str, uuid: str, owner: Union[str, None]=None, tags: Uni
c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?, ?, ?, ?, ?, datetime('now'), datetime('now'))", (uuid, alias_str, tag_str, owner, code))
conn.commit()
conn.close()
dl.save(path + "/backlogger.db", message="Added entry for project " + uuid + " to database", dataset=path)
dl.save(db, message="Added entry for project " + uuid + " to database", dataset=path)
def update_project_data(db, uuid, prop, value = None):
def update_project_data(path, db, uuid, prop, value = None):
dl.get(db, dataset=path)
conn = sqlite3.connect(db)
c = conn.cursor()
c.execute(f"UPDATE projects SET '{prop}' = '{value}' WHERE id == '{uuid}'")
@ -54,6 +57,7 @@ def update_project_data(db, uuid, prop, value = None):
def update_aliases(path: str, uuid: str, aliases: list[str]):
db = os.path.join(path, "backlogger.db")
dl.get(db, dataset=path)
known_data = _project_lookup_by_id(db, uuid)[0]
known_aliases = known_data[1]
@ -117,11 +121,13 @@ def import_project(path: str, url: str, owner: Union[str, None]=None, tags: Unio
if not uuid:
raise ValueError("The dataset does not have a uuid!")
if not os.path.exists(path + "/projects/" + uuid):
dl.unlock(path + "/backlogger.db", dataset=path)
db = path + "/backlogger.db"
dl.get(db, ds=path)
dl.unlock(db, dataset=path)
create_project(path, uuid, owner, tags, aliases, code)
move_submodule(path, 'projects/tmp', 'projects/' + uuid)
os.mkdir(path + '/import_scripts/' + uuid)
dl.save([path + "/backlogger.db", path + '/projects/' + uuid], message="Import project from " + url, dataset=path)
dl.save([db, path + '/projects/' + uuid], message="Import project from " + url, dataset=path)
else:
dl.drop(tmp_path, reckless='kill')
shutil.rmtree(tmp_path)