change get_url to return first url string

1 files changed, 24 insertions, 15 deletions

diff --git a/bjoern/videoanalyse/read_sqlite.py b/bjoern/videoanalyse/read_sqlite.py
index 2041ac3..6c3e5b0 100644
--- a/bjoern/videoanalyse/read_sqlite.py
+++ b/bjoern/videoanalyse/read_sqlite.py
@@ -1,32 +1,41 @@
-import sqlite3;
-import fileinput;
-import csv;
+import sqlite3
+import csv
+
 
 def print_sqlite(cursor):
     for row in cursor.fetchall():
         print(dict(row))
 
+
 def create_csv_from_sqlite(data):
-    with open('sqlite_output.csv', 'w', newline='') as f:
+    with open("sqlite_output.csv", "w", newline="") as f:
         writer = csv.writer(f)
-        writer.writerow(['id', 'url', ...])
+        writer.writerow(["id", "url", ...])
         writer.writerows(data)
-        
+
+
 def get_url_from_sqlite(sqlite_file, title):
     con = sqlite3.connect(sqlite_file)
     con.row_factory = sqlite3.Row
-    
+
     cur = con.cursor()
-    select_statement = 'SELECT url FROM moz_places WHERE title = ?'
+    select_statement = "SELECT url FROM moz_places WHERE title = ?"
     dat = cur.execute(select_statement, (title,))
-    
-    print_sqlite(cur) 
-        
+
+    row = dat.fetchone()
+    if row:
+        return row["url"]
+    return None
+
+
 def main():
-    get_url_from_sqlite('chsokl11/places.sqlite', 'Atmosphärische Gegenstrahlung – Wikipedia')
-    
-    #create_csv_from_sqlite(dat)
-    #concat_logfiles()   
+    get_url_from_sqlite(
+        "chsokl11/places.sqlite", "Atmosphärische Gegenstrahlung – Wikipedia"
+    )
+
+    # create_csv_from_sqlite(dat)
+    # concat_logfiles()
+
 
 if __name__ == "__main__":
     main()