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)
);""")
self.dbCursor.execute("""CREATE TABLE IF NOT EXISTS starFitType
self.dbCursor.execute("""CREATE TABLE IF NOT EXISTS starFoldedFitType
(
mainName TEXT,
fitType TEXT,
foldedFitType TEXT,
PRIMARY KEY (mainName),
FOREIGN KEY (mainName) REFERENCES starnames(mainName)
);""")
@@ -110,9 +110,9 @@ class StarDB():
self.dbCursor.execute("""UPDATE stars
SET filename = replace(filename, '\\', '/')
WHERE filename LIKE '%\\%';""")
self.dbCursor.execute("""UPDATE starFitType
SET fitType = 'sine'
WHERE fitType IS NULL OR fitType = '';""")
self.dbCursor.execute("""UPDATE starFoldedFitType
SET foldedFitType = 'sine'
WHERE foldedFitType IS NULL OR foldedFitType = '';""")
def insertDefaultValues(self):
@@ -154,9 +154,9 @@ class StarDB():
self.connection.commit()
def updateFitType(self, mainName, fitType: str):
if(fitType in supportedFitTypes):
self.dbCursor.execute(f"""UPDATE starFitType
SET fitType = '{fitType}'
if(fitType in supportedFoldedFitTypes):
self.dbCursor.execute(f"""UPDATE starFoldedFitType
SET foldedFitType = '{fitType}'
WHERE mainName = '{mainName}';""")
self.connection.commit()
else:
@@ -211,11 +211,11 @@ class StarDB():
""")
return res.fetchone()[0]
def getFitType(self, mainName):
res = self.dbCursor.execute(f"""SELECT mainName FROM starFitType
def getFoldedFitType(self, mainName):
res = self.dbCursor.execute(f"""SELECT foldedFitType FROM starFoldedFitType
WHERE mainName = \"{mainName}\";""")
ret = res.fetchone()[0]
if(ret in supportedFitTypes):
if(ret in supportedFoldedFitTypes):
return ret
else:
return "sine"