Started python script
This commit is contained in:
parent
ae9f4e4497
commit
d324ddb3c9
2 changed files with 60 additions and 0 deletions
1
requirements.txt
Normal file
1
requirements.txt
Normal file
|
@ -0,0 +1 @@
|
|||
tabulate==0.8.10
|
59
start.py
Normal file
59
start.py
Normal file
|
@ -0,0 +1,59 @@
|
|||
import argparse
|
||||
import sqlite3
|
||||
from tabulate import tabulate
|
||||
|
||||
def parse_args():
|
||||
"""
|
||||
Parses the arguments from the CLI
|
||||
"""
|
||||
parser = argparse.ArgumentParser(description="Testtool Adress Changes")
|
||||
parser.add_argument(
|
||||
'-c', '--create',
|
||||
help='Creates an example database',
|
||||
action='store_true'
|
||||
)
|
||||
parser.add_argument(
|
||||
'-i', '--insert',
|
||||
help='insert some example data',
|
||||
action='store_true'
|
||||
)
|
||||
parser.add_argument(
|
||||
'-p', '--print',
|
||||
choices=['table', 'string'],
|
||||
default='table',
|
||||
help='prints all the cells from the table',
|
||||
)
|
||||
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def execute_sql(args):
|
||||
db = sqlite3.connect("test.db")
|
||||
cur = db.cursor()
|
||||
print(args.print)
|
||||
if args.create:
|
||||
print("Creating")
|
||||
with open ('sql/create_db.sql', 'r') as sql_file:
|
||||
cur.executescript(sql_file.read())
|
||||
elif args.insert:
|
||||
print("insert")
|
||||
with open ('sql/insert_data.sql', 'r') as sql_file:
|
||||
cur.executescript(sql_file.read())
|
||||
elif args.print:
|
||||
cur.execute("SELECT * from customers")
|
||||
if args.print == 'table':
|
||||
print(tabulate(cur.fetchall()))
|
||||
else:
|
||||
print(cur.fetchall())
|
||||
db.commit()
|
||||
db.close()
|
||||
|
||||
|
||||
|
||||
def main():
|
||||
args = parse_args()
|
||||
execute_sql(args)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
Loading…
Reference in a new issue