forked from importers/git-importer
87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
import psycopg2
|
|
from config import config
|
|
|
|
|
|
class DB:
|
|
def __init__(self):
|
|
self.connect()
|
|
|
|
def connect(self):
|
|
try:
|
|
# read the connection parameters
|
|
params = config()
|
|
# connect to the PostgreSQL server
|
|
self.conn = psycopg2.connect(**params)
|
|
|
|
except (Exception, psycopg2.DatabaseError) as error:
|
|
print(error)
|
|
raise error
|
|
|
|
def schema_version(self):
|
|
# create a cursor
|
|
cur = self.conn.cursor()
|
|
|
|
# execute a statement
|
|
try:
|
|
cur.execute("SELECT MAX(version) from scheme")
|
|
except psycopg2.errors.UndefinedTable as error:
|
|
cur.close()
|
|
self.close()
|
|
self.connect()
|
|
return 0
|
|
|
|
db_version = cur.fetchone()
|
|
|
|
cur.close()
|
|
return db_version[0]
|
|
|
|
def close(self):
|
|
if self.conn is not None:
|
|
self.conn.close()
|
|
self.conn = None
|
|
|
|
def create_tables(self):
|
|
"""Create the tables if not existing - assumes connected"""
|
|
|
|
commands = (
|
|
"""
|
|
CREATE TABLE scheme (
|
|
id SERIAL PRIMARY KEY,
|
|
version SMALLINT NOT NULL
|
|
)
|
|
""",
|
|
"INSERT INTO scheme (version) VALUES(1)",
|
|
""" CREATE TABLE revisions (
|
|
id SERIAL PRIMARY KEY,
|
|
project VARCHAR(255) NOT NULL,
|
|
package VARCHAR(255) NOT NULL,
|
|
rev VARCHAR(255) NOT NULL,
|
|
unexpanded_srcmd5 VARCHAR(255) NOT NULL,
|
|
commit_time timestamp NOT NULL,
|
|
userid VARCHAR(255) NOT NULL,
|
|
comment VARCHAR(255)
|
|
)
|
|
""",
|
|
)
|
|
|
|
if self.schema_version() > 0:
|
|
return
|
|
try:
|
|
cur = self.conn.cursor()
|
|
# create table one by one
|
|
for command in commands:
|
|
cur.execute(command)
|
|
# close communication with the PostgreSQL database server
|
|
cur.close()
|
|
# commit the changes
|
|
self.conn.commit()
|
|
except (Exception, psycopg2.DatabaseError) as error:
|
|
print(error)
|
|
self.close()
|
|
raise error
|
|
|
|
|
|
if __name__ == "__main__":
|
|
db = DB()
|
|
db.create_tables()
|