Skip to content

Commit 8dc1ec5

Browse files
Update to using generator and utilize lower-case indexing in admin.py
1 parent 79d61fb commit 8dc1ec5

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

datajoint/admin.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,11 @@ def kill(restriction=None, connection=None, order_by=None): # pragma: no cover
4949
while True:
5050
print(' ID USER HOST STATE TIME INFO')
5151
print('+--+ +----------+ +-----------+ +-----------+ +-----+')
52-
cur = [{k.upper(): v for k, v in elem.items()}
53-
for elem in connection.query(query, as_dict=True)]
52+
cur = ({k.lower(): v for k, v in elem.items()}
53+
for elem in connection.query(query, as_dict=True))
5454
for process in cur:
5555
try:
56-
print('{ID:>4d} {USER:<12s} {HOST:<12s} {STATE:<12s} {TIME:>7d} {INFO}'.format(**process))
56+
print('{id:>4d} {user:<12s} {host:<12s} {state:<12s} {time:>7d} {info}'.format(**process))
5757
except TypeError:
5858
print(process)
5959
response = input('process to kill or "q" to quit > ')
@@ -89,10 +89,10 @@ def kill_quick(restriction=None, connection=None):
8989
query = 'SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()' + (
9090
"" if restriction is None else ' AND (%s)' % restriction)
9191

92-
cur = [{k.upper(): v for k, v in elem.items()}
93-
for elem in connection.query(query, as_dict=True)]
92+
cur = ({k.lower(): v for k, v in elem.items()}
93+
for elem in connection.query(query, as_dict=True))
9494
nkill = 0
9595
for process in cur:
96-
connection.query('kill %d' % process['ID'])
96+
connection.query('kill %d' % process['id'])
9797
nkill += 1
9898
return nkill

datajoint/dependencies.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,15 @@ def load(self):
4343
self.add_node(n, primary_key=pk)
4444

4545
# load foreign keys
46-
keys = [{k.lower(): v for k, v in elem.items()} for elem in self._conn.query("""
46+
keys = ({k.lower(): v for k, v in elem.items()} for elem in self._conn.query("""
4747
SELECT constraint_name,
4848
concat('`', table_schema, '`.`', table_name, '`') as referencing_table,
4949
concat('`', referenced_table_schema, '`.`', referenced_table_name, '`') as referenced_table,
5050
column_name, referenced_column_name
5151
FROM information_schema.key_column_usage
5252
WHERE referenced_table_name NOT LIKE "~%%" AND (referenced_table_schema in ('{schemas}') OR
5353
referenced_table_schema is not NULL AND table_schema in ('{schemas}'))
54-
""".format(schemas="','".join(self._conn.schemas)), as_dict=True)]
54+
""".format(schemas="','".join(self._conn.schemas)), as_dict=True))
5555
fks = defaultdict(lambda: dict(attr_map=OrderedDict()))
5656
for key in keys:
5757
d = fks[(key['constraint_name'], key['referencing_table'], key['referenced_table'])]

datajoint/external.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,11 +276,11 @@ def references(self):
276276
"""
277277
:return: generator of referencing table names and their referencing columns
278278
"""
279-
return [{k.lower(): v for k, v in elem.items()} for elem in self.connection.query("""
279+
return ({k.lower(): v for k, v in elem.items()} for elem in self.connection.query("""
280280
SELECT concat('`', table_schema, '`.`', table_name, '`') as referencing_table, column_name
281281
FROM information_schema.key_column_usage
282282
WHERE referenced_table_name="{tab}" and referenced_table_schema="{db}"
283-
""".format(tab=self.table_name, db=self.database), as_dict=True)]
283+
""".format(tab=self.table_name, db=self.database), as_dict=True))
284284

285285
def fetch_external_paths(self, **fetch_kwargs):
286286
"""

0 commit comments

Comments
 (0)