From 88bec489c05e2e7df7b836e6974b17d9d6f5b36d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Victor=20Frans=20Pondaco=20Winandy?=
<jvfpw18@inf.ufpr.br>
Date: Thu, 23 Jan 2020 08:35:23 -0300
Subject: [PATCH] Fix multiple columns in pk
---
database/database_table.py | 11 ++++-------
1 file changed, 4 insertions(+), 7 deletions(-)
diff --git a/database/database_table.py b/database/database_table.py
index 579619a..ed495b6 100644
--- a/database/database_table.py
+++ b/database/database_table.py
@@ -201,14 +201,11 @@ class DatabaseTable(Table):
bind = self.metadata.bind
pks = get_primary_keys(self)
- primary_keys = []
- for pk in pks:
- primary_keys.append(ttable.columns.get(pk.name))
+ primary_keys = (ttable.columns.get(pk.name) for pk in pks)
- for pk in primary_keys:
- new_pk = PrimaryKeyConstraint(pk)
- query = AddConstraint(new_pk)
- bind.execute(query)
+ new_pk = PrimaryKeyConstraint(*primary_keys)
+ query = AddConstraint(new_pk)
+ bind.execute(query)
def populate_temporary(self, ttable, in_file, header, year, delimiters=[';', '\\n', '"'],
null='', offset=2, bind=None):
--
GitLab