lib/exercises-in-programming-style
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Let's write the DB on a file already...

Browse Source
This commit is contained in:
Crista Lopes
2014-01-01 15:28:47 -08:00
parent ff435c3351
commit 636b98be97
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
24-tabular/tf-24.py
View File

@@ -1,5 +1,5 @@
#!/usr/bin/env python #!/usr/bin/env python
import sys, re, string, sqlite3 import sys, re, string, sqlite3, os.path
# #
# The relational database of this problem consists of 3 tables: # The relational database of this problem consists of 3 tables:
@@ -52,18 +52,18 @@ def load_file_into_database(path_to_file, connection):
c.close() c.close()
# #
# The main function # Create if it doesn't exist
# #
connection = sqlite3.connect(':memory:') if not os.path.isfile('tf.db'):
create_db_schema(connection) with sqlite3.connect('tf.db') as connection:
load_file_into_database(sys.argv[1], connection) create_db_schema(connection)
load_file_into_database(sys.argv[1], connection)
# Now, let's query # Now, let's query
c = connection.cursor() with sqlite3.connect('tf.db') as connection:
c.execute("SELECT value, COUNT(*) as C FROM words GROUP BY value ORDER BY C DESC") c = connection.cursor()
for i in range(25): c.execute("SELECT value, COUNT(*) as C FROM words GROUP BY value ORDER BY C DESC")
row = c.fetchone() for i in range(25):
if row != None: row = c.fetchone()
print row[0] + ' - ' + str(row[1]) if row != None:
print row[0] + ' - ' + str(row[1])
connection.close()