1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
#!/usr/bin/python
import csv
import psycopg2
import bz2
import ConfigParser
def main(config, out):
# Setup database connection
c = ConfigParser.ConfigParser({'user': '', 'password': ''})
c.read(config)
dbname = c.get('database', 'name')
dbuser = c.get('database', 'user')
dbpass = c.get('database', 'password')
dbhost = c.get('database', 'host')
dbconn = psycopg2.connect(database=dbname, user=dbuser, password=dbpass,
host=dbhost)
cursor = dbconn.cursor()
cursor.execute('select name,version,filename,downloads from release_files')
with bz2.BZ2File(out, 'w') as f:
w = csv.writer(f)
w.writerows(cursor.fetchall())
dbconn.close()
if __name__ == '__main__':
import sys
main(sys.argv[1], sys.argv[2])
|