Reduce complexity of __getitem__ API

Browse Source 
If you need to specify more, call get_table method
This commit is contained in:
Stefan Wehrmeyer 2014-01-31 19:18:58 +01:00
parent 7a6b16931c
commit 0e39190062
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dataset/persistence/database.py
View File

@ -223,10 +223,7 @@ class Database(object):
self._release()
def __getitem__(self, table_name):
if type(table_name) is tuple:
return self.get_table(*table_name[:3])
else:
return self.get_table(table_name)
return self.get_table(table_name)
def query(self, query, **kw):
"""

6
test/test_persistence.py
View File

@ -76,7 +76,7 @@ class DatabaseTestCase(unittest.TestCase):
def test_create_table_shorthand1(self):
pid = "int_id"
table = self.db['foo5', pid]
table = self.db.get_table('foo5', pid)
assert table.table.exists
assert len(table.table.columns) == 1, table.table.columns
assert pid in table.table.c, table.table.c
@ -89,7 +89,7 @@ class DatabaseTestCase(unittest.TestCase):
def test_create_table_shorthand2(self):
pid = "string_id"
table = self.db['foo6', pid, 'String']
table = self.db.get_table('foo6', primary_id=pid, primary_type='String')
assert table.table.exists
assert len(table.table.columns) == 1, table.table.columns
assert pid in table.table.c, table.table.c
@ -100,7 +100,7 @@ class DatabaseTestCase(unittest.TestCase):
def test_create_table_shorthand3(self):
pid = "string_id"
table = self.db['foo7', pid, 'String(20)']
table = self.db.get_table('foo7', primary_id=pid, primary_type='String(20)')
assert table.table.exists
assert len(table.table.columns) == 1, table.table.columns
assert pid in table.table.c, table.table.c