40 lines
929 B
Python
40 lines
929 B
Python
import configparser
|
|
import psycopg2
|
|
from sql_queries import create_table_queries, drop_table_queries
|
|
|
|
|
|
def drop_tables(cur, conn):
|
|
for query in drop_table_queries:
|
|
try:
|
|
cur.execute(query)
|
|
conn.commit()
|
|
print(query + ' - Done!')
|
|
except:
|
|
print(query + ' - Fail!')
|
|
pass
|
|
|
|
def create_tables(cur, conn):
|
|
for query in create_table_queries:
|
|
try:
|
|
cur.execute(query)
|
|
conn.commit()
|
|
print(query + ' - Done!')
|
|
except:
|
|
print(query + ' - Fail!')
|
|
pass
|
|
|
|
def main():
|
|
config = configparser.ConfigParser()
|
|
config.read('dwh.cfg')
|
|
|
|
conn = psycopg2.connect("host={} dbname={} user={} password={} port={}".format(*config['CLUSTER'].values()))
|
|
cur = conn.cursor()
|
|
|
|
drop_tables(cur, conn)
|
|
create_tables(cur, conn)
|
|
|
|
conn.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |