added conventions

Author: AmirPupko

example_runner.py

@@ -1,7 +1,7 @@
 from copy import copy
 import pandas_to_sql
 from pandas_to_sql.testing.utils.fake_data_creation import create_fake_dataset
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
+from pandas_to_sql.conventions import flatten_grouped_dataframe
 
 table_name = 'random_data'
 df, _ = create_fake_dataset()

pandas_to_sql/conventions/__init__.py

@@ -0,0 +1 @@
+from pandas_to_sql.conventions.groupby_conventions import flatten_grouped_dataframe

pandas_to_sql/conventions/groupby_conventions.py

@@ -0,0 +1,17 @@
+import pandas as pd
+from pandas_to_sql.utils.pandas_dataframe_intercepter import PandasDataFrameIntercepter
+from copy import copy
+
+def flatten_grouped_dataframe(df):
+    if not isinstance(df, PandasDataFrameIntercepter):
+        raise Exception(f"can only get type {str(type(PandasDataFrameIntercepter))}")
+    
+    df_c = copy(df.df_pandas)
+    if isinstance(df_c, pd.core.series.Series):
+        series_name = df_c.name
+        new_col_name = list(filter(lambda k: k.startswith(series_name), df.df_sql_convert_table.columns.keys()))[0]
+        df_c = df_c.reset_index().rename(columns={series_name: new_col_name})
+    else:
+        df_c.columns = df_c.columns.map('_'.join)
+        df_c = df_c.reset_index()
+    return PandasDataFrameIntercepter(df_c, copy(df.df_sql_convert_table))

pandas_to_sql/testing/tests/test_concat.py

@@ -1,6 +1,6 @@
 import pytest
 from pandas_to_sql.testing.utils.asserters import assert_
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
+from pandas_to_sql.conventions import flatten_grouped_dataframe
 from copy import copy
 import pandas as pd
 import pandas_to_sql 

pandas_to_sql/testing/tests/test_datetime.py

@@ -1,7 +1,6 @@
 from datetime import timedelta, datetime
 import pytest
 from pandas_to_sql.testing.utils.asserters import assert_, get_expected_and_actual
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
 from copy import copy
 import pandas as pd
 import pandas_to_sql 

pandas_to_sql/testing/tests/test_groupby.py

@@ -1,6 +1,6 @@
 import pytest
 from pandas_to_sql.testing.utils.asserters import assert_
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
+from pandas_to_sql.conventions import flatten_grouped_dataframe
 
 
 

pandas_to_sql/testing/tests/test_merge.py

@@ -1,6 +1,6 @@
 import pytest
 from pandas_to_sql.testing.utils.asserters import assert_
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
+from pandas_to_sql.conventions import flatten_grouped_dataframe
 from copy import copy
 
 

pandas_to_sql/testing/tests/test_str.py

@@ -1,7 +1,6 @@
 from datetime import timedelta, datetime
 import pytest
 from pandas_to_sql.testing.utils.asserters import assert_, get_expected_and_actual
-from pandas_to_sql.utils.helpers import flatten_grouped_dataframe
 from copy import copy
 import pandas as pd
 import pandas_to_sql 

pandas_to_sql/utils/helpers.py

@@ -1,23 +1,4 @@
 import pandas as pd
-from pandas_to_sql.engine.table import Table
-from pandas_to_sql.engine.grouped_table import GroupedTable
-from pandas_to_sql.utils.pandas_dataframe_intercepter import PandasDataFrameIntercepter
-from copy import copy
-
-## Conventions
-def flatten_grouped_dataframe(df):
-    if not isinstance(df, PandasDataFrameIntercepter):
-        raise Exception(f"can only get type {str(type(PandasDataFrameIntercepter))}")
-    
-    df_c = copy(df.df_pandas)
-    if isinstance(df_c, pd.core.series.Series):
-        series_name = df_c.name
-        new_col_name = list(filter(lambda k: k.startswith(series_name), df.df_sql_convert_table.columns.keys()))[0]
-        df_c = df_c.reset_index().rename(columns={series_name: new_col_name})
-    else:
-        df_c.columns = df_c.columns.map('_'.join)
-        df_c = df_c.reset_index()
-    return PandasDataFrameIntercepter(df_c, copy(df.df_sql_convert_table))
 
 
 ## Types