diff --git a/gethandles.py b/gethandles.py index f246c10..45c5828 100755 --- a/gethandles.py +++ b/gethandles.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- -import operator, re, sqlite3 +import config, operator, re, sqlite3 -def getTweets(database_filename = "Database.db"): +def getTweets(database_filename = config.dbpath): sql_conn = sqlite3.connect(database_filename) cur = sql_conn.cursor() @@ -23,6 +23,5 @@ def getTweets(database_filename = "Database.db"): return handles if __name__ == "__main__": - data = sorted(list(getTweets().items()), key=operator.itemgetter(1), reverse=True) - for handle, tweets in data: + for handle, tweets in sorted(list(getTweets().items()), key=operator.itemgetter(1), reverse=True): print(handle + "," + str(tweets)) diff --git a/makedb.py b/makedb.py index 1a0f271..712ab8a 100755 --- a/makedb.py +++ b/makedb.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- -import os, sqlite3, sys +import config, os, sqlite3, sys -def makeDB(path="Database.db"): +def makeDB(path=config.dbpath): if os.path.isfile(path): raise IOError(path + " already exists. If you want to recreate it, please delete it first, or provide a different file name.")