comparison unique_mlg_id.py @ 1:6c10123cb4a7 draft default tip

Uploaded
author greg
date Wed, 11 Dec 2024 16:28:27 +0000
parents c39c543605a2
children
comparison
equal deleted inserted replaced
0:c39c543605a2 1:6c10123cb4a7
2 import argparse 2 import argparse
3 import sys 3 import sys
4 4
5 import psycopg2 5 import psycopg2
6 6
7 from sqlalchemy import MetaData, create_engine 7 from sqlalchemy import create_engine
8 from sqlalchemy.engine.url import make_url 8 from sqlalchemy.engine.url import make_url
9 9
10 10
11 class UniqueMGLIDGenerator(object): 11 class UniqueMGLIDGenerator(object):
12 def __init__(self): 12 def __init__(self):
14 self.conn = None 14 self.conn = None
15 self.parse_args() 15 self.parse_args()
16 self.outfh = open(self.args.output, "w") 16 self.outfh = open(self.args.output, "w")
17 self.connect_db() 17 self.connect_db()
18 self.engine = create_engine(self.args.database_connection_string) 18 self.engine = create_engine(self.args.database_connection_string)
19 self.metadata = MetaData(self.engine)
20 19
21 def parse_args(self): 20 def parse_args(self):
22 parser = argparse.ArgumentParser() 21 parser = argparse.ArgumentParser()
23 parser.add_argument('--database_connection_string', dest='database_connection_string', help='Postgres database connection string'), 22 parser.add_argument('--database_connection_string', dest='database_connection_string', help='Postgres database connection string'),
24 parser.add_argument('--output', dest='output', help='Output dataset'), 23 parser.add_argument('--output', dest='output', help='Output dataset'),