From b6b5918493260518ee4b868ea03e5316caca40aa Mon Sep 17 00:00:00 2001 From: Niclas Dobbertin Date: Thu, 15 Jun 2023 16:05:04 +0200 Subject: change get_url to return first url string --- bjoern/videoanalyse/read_sqlite.py | 39 +++++++++++++++++++++++--------------- 1 file changed, 24 insertions(+), 15 deletions(-) (limited to 'bjoern/videoanalyse/read_sqlite.py') 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() -- cgit v1.2.3