Skip to content

Added basic list_tables function for schema class #844

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
dimitri-yatsenko merged 7 commits into from
Dec 11, 2020
Merged

Added basic list_tables function for schema class #844

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
b3e7570
Added basic list_tables function for schema class
Synicix Dec 9, 2020
32b2874
Update list_tables to more efficient solution
Synicix Dec 9, 2020
804ba11
Update comments to fit in 95 characters convention
Synicix Dec 9, 2020
8288073
Added test for schema.list_tables
Synicix Dec 10, 2020
2d921a5
Merge remote-tracking branch 'upstream/master'
Synicix Dec 10, 2020
82d23a7
Format line to fit is 95 char requirements...
Synicix Dec 10, 2020
9021ff9
Remove white space
Synicix Dec 10, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions datajoint/schemas.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -288,6 +288,17 @@ def repl(s):
with open(python_filename, 'wt') as f:
f.write(python_code)

def list_tables(self):
"""
Return a list of all tables in the schema except tables with ~ in first character such
as ~logs and ~job
:return: A list of table names in their raw datajoint naming convection form
"""

return [table_name for (table_name,) in self.connection.query("""
SELECT table_name FROM information_schema.tables
WHERE table_schema = %s and table_name NOT LIKE '~%%'""", args=(self.database))]


class VirtualModule(types.ModuleType):
"""
Expand Down
6 changes: 6 additions & 0 deletions tests/test_schema.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from . import schema
from . import schema_empty
from . import PREFIX, CONN_INFO
from .schema_simple import schema as schema_simple


def relation_selector(attr):
Expand Down Expand Up @@ -105,6 +106,11 @@ class Unit(dj.Part):

test_schema.drop()

def test_list_tables():
assert(['#a', '#argmax_test', '#data_a', '#data_b', '#i_j', '#j_i', '#l',\
'#t_test_update', '__b', '__b__c', '__d', '__e', '__e__f', 'f',\
'reserved_word'] == schema_simple.list_tables())


def test_schema_save():
assert_true("class Experiment(dj.Imported)" in schema.schema.code)
Expand Down