Skip to content

Commit 749b5bf

Browse files
committed
export properly, fix extra paren
1 parent f31378e commit 749b5bf

File tree

3 files changed

+15
-1
lines changed

3 files changed

+15
-1
lines changed

osc_ingest_trino/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,13 @@
77
# defines the release version for this python package
88
__version__ = "0.1.0"
99

10+
from .sqlcols import *
11+
from .sqltypes import *
12+
1013
__all__ = [
1114
"sql_compliant_name",
1215
"enforce_sql_column_names",
1316
"pandas_type_to_sql",
1417
"create_table_schema_pairs",
1518
]
19+

osc_ingest_trino/sqlcols.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,11 @@
11
import re
22
import pandas as pd
33

4+
__all__ = [
5+
"sql_compliant_name",
6+
"enforce_sql_column_names",
7+
]
8+
49
_wsdedup = re.compile(r"\s+")
510
_usdedup = re.compile(r"__+")
611
_rmpunc = re.compile(r"[,.()&$/+-]+")
@@ -36,6 +41,6 @@ def enforce_sql_column_names(df, inplace=False, maxlen=63):
3641
ocols = sql_compliant_name(icols, maxlen=maxlen)
3742
if (len(set(ocols)) < len(ocols)):
3843
raise ValueError("remapped column names were not unique!")
39-
rename_map = dict(list(zip(icols, ocols))))
44+
rename_map = dict(list(zip(icols, ocols)))
4045
return df.rename(columns=rename_map, inplace=inplace)
4146

osc_ingest_trino/sqltypes.py

+5
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
11
import pandas as pd
22

3+
__all__ = [
4+
"pandas_type_to_sql",
5+
"create_table_schema_pairs",
6+
]
7+
38
_p2smap = {
49
'string': 'varchar',
510
'float32': 'real',

0 commit comments

Comments
 (0)