diff --git a/src/model.py b/src/model.py index 294ab81..a3bf243 100755 --- a/src/model.py +++ b/src/model.py @@ -21,7 +21,12 @@ def generate_company_list() -> Tuple[list, list]: :return: (list) A list of companies. """ - conn = psycopg2.connect(host="172.19.0.2", database="stocks", user="postgres", password="123456", port="5432") + conn = psycopg2.connect( + host="172.19.0.2", + database="stocks", + user="postgres", + password="123456", + port="5432") cursor = conn.cursor() cursor.execute("SELECT * FROM companies;") records = cursor.fetchall()