Merge pull request #35 from 3onyc/issue-33

Browse Source 
Use string comparison in create_table (Fixes issue #33)
This commit is contained in:
Friedrich Lindenberg 2013-11-13 02:07:25 -08:00
commit 4856c16b1f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dataset/persistence/database.py
View File

@ -121,12 +121,12 @@ class Database(object):
try: try:
log.debug("Creating table: %s on %r" % (table_name, self.engine)) log.debug("Creating table: %s on %r" % (table_name, self.engine))
table = SQLATable(table_name, self.metadata) table = SQLATable(table_name, self.metadata)
if primary_type is 'Integer': if primary_type == 'Integer':
auto_flag = False auto_flag = False
if primary_id is 'id': if primary_id == 'id':
auto_flag = True auto_flag = True
col = Column(primary_id, Integer, primary_key=True, autoincrement=auto_flag) col = Column(primary_id, Integer, primary_key=True, autoincrement=auto_flag)
elif primary_type is 'Text': elif primary_type == 'Text':
col = Column(primary_id, Text, primary_key=True) col = Column(primary_id, Text, primary_key=True)
else: else:
raise DatasetException( raise DatasetException(