SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

StarsDB: make naming for fitType clearer

Browse Source
This commit is contained in:
SGCMarkus
2024-08-19 11:53:29 +02:00
parent 15dce775d2
commit e3a8575456
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/astrodatagui/db/StarsDB.py
+11 -11
View File
@@ -89,10 +89,10 @@ class StarDB():
FOREIGN KEY (mainName) REFERENCES starnames(mainName) FOREIGN KEY (mainName) REFERENCES starnames(mainName)
);""") );""")
self.dbCursor.execute("""CREATE TABLE IF NOT EXISTS starFitType self.dbCursor.execute("""CREATE TABLE IF NOT EXISTS starFoldedFitType
( (
mainName TEXT, mainName TEXT,
fitType TEXT, foldedFitType TEXT,
PRIMARY KEY (mainName), PRIMARY KEY (mainName),
FOREIGN KEY (mainName) REFERENCES starnames(mainName) FOREIGN KEY (mainName) REFERENCES starnames(mainName)
);""") );""")
@@ -110,9 +110,9 @@ class StarDB():
self.dbCursor.execute("""UPDATE stars self.dbCursor.execute("""UPDATE stars
SET filename = replace(filename, '\\', '/') SET filename = replace(filename, '\\', '/')
WHERE filename LIKE '%\\%';""") WHERE filename LIKE '%\\%';""")
self.dbCursor.execute("""UPDATE starFitType self.dbCursor.execute("""UPDATE starFoldedFitType
SET fitType = 'sine' SET foldedFitType = 'sine'
WHERE fitType IS NULL OR fitType = '';""") WHERE foldedFitType IS NULL OR foldedFitType = '';""")
def insertDefaultValues(self): def insertDefaultValues(self):
@@ -154,9 +154,9 @@ class StarDB():
self.connection.commit() self.connection.commit()
def updateFitType(self, mainName, fitType: str): def updateFitType(self, mainName, fitType: str):
if(fitType in supportedFitTypes): if(fitType in supportedFoldedFitTypes):
self.dbCursor.execute(f"""UPDATE starFitType self.dbCursor.execute(f"""UPDATE starFoldedFitType
SET fitType = '{fitType}' SET foldedFitType = '{fitType}'
WHERE mainName = '{mainName}';""") WHERE mainName = '{mainName}';""")
self.connection.commit() self.connection.commit()
else: else:
@@ -211,11 +211,11 @@ class StarDB():
""") """)
return res.fetchone()[0] return res.fetchone()[0]
def getFitType(self, mainName): def getFoldedFitType(self, mainName):
res = self.dbCursor.execute(f"""SELECT mainName FROM starFitType res = self.dbCursor.execute(f"""SELECT foldedFitType FROM starFoldedFitType
WHERE mainName = \"{mainName}\";""") WHERE mainName = \"{mainName}\";""")
ret = res.fetchone()[0] ret = res.fetchone()[0]
if(ret in supportedFitTypes): if(ret in supportedFoldedFitTypes):
return ret return ret
else: else:
return "sine" return "sine"