Skip to content

Commit 620832d

Browse files
committed
add type check
1 parent 94cf2ca commit 620832d

1 file changed

Lines changed: 22 additions & 2 deletions

File tree

pdb2sql/many2sql.py

Lines changed: 22 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,30 @@
1313
class many2sql(pdb2sql):
1414

1515
def __init__(self, pdbfiles, tablenames=None):
16-
"""Create a sql database containing multiple pdbs."""
16+
"""Create a sql database containing multiple pdbs.
17+
18+
Args:
19+
pdbfiles(list): list of pdb files or data.
20+
tablenames (list): list of table names in string
21+
Defaults to None.
22+
23+
Raises:
24+
TypeError: pdbfiles must be a list
25+
TypeError: tablenames must be a list
26+
TypeError: tablenames must be a list of strings
27+
"""
1728

1829
if not isinstance(pdbfiles, list):
19-
raise ValueError('pdbfiles must be a list')
30+
raise TypeError('pdbfiles must be a list')
31+
32+
if tablenames is not None:
33+
if not isinstance(tablenames, list):
34+
raise TypeError('tablenames must be a list')
35+
else:
36+
for i in tablenames:
37+
if not isinstance(i, str):
38+
raise TypeError(f'tablenames must be a list of strings,'
39+
f' {i} is a type of {type(i)}')
2040

2141
self.npdb = len(pdbfiles)
2242

0 commit comments

Comments
 (0)