Refactor.

Author: toddbirchard

main.py

@@ -1,5 +1,5 @@
 """App entry point."""
-from tutorial import init_script
+from pandas_sqlalchemy_tutorial import init_script
 
 script = init_script()
 

tutorial/__init__.py renamed to pandas_sqlalchemy_tutorial/__init__.py

@@ -1,16 +1,17 @@
 """Fetch and insert data into SQL database using Pandas."""
-from .read import load_csv_data
-from .database import Database
 from config import (
 	SQLALCHEMY_DATABASE_URI,
 	SQLALCHEMY_ENGINE_OPTIONS,
 	CSV_FILE
 )
+from .read import load_csv_data
+from .database import Database
 
 
 def init_script():
 	"""Demonstrate Pandas' SQLAlchemy integration."""
 	jobs_df = load_csv_data(CSV_FILE)
 	db = Database(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_ENGINE_OPTIONS)
-	db.upload_dataframe_to_sql(jobs_df, 'nyc_jobs')
-	db.get_dataframe_from_sql('nyc_jobs')
+	upload_result = db.upload_dataframe_to_sql(jobs_df, 'nyc_jobs')
+	download_result = db.get_dataframe_from_sql('nyc_jobs')
+	return upload_result, download_result

tutorial/database.py renamed to pandas_sqlalchemy_tutorial/database.py

@@ -37,15 +37,16 @@ def upload_dataframe_to_sql(self, csv_df, table_name):
                 "posting_updated": DateTime
             }
         )
-        print(f'Loaded {len(csv_df)} rows into {table_name} table.')
+        result = f'Loaded {len(csv_df)} rows INTO {table_name} table.'
+        print(result)
+        return result
 
     def get_dataframe_from_sql(self, table_name):
         """Create DataFrame form SQL table."""
         table_df = pd.read_sql_table(
             table_name,
             con=self.engine
         )
-        print(f'Loaded {len(table_df)} rows from {table_name}.')
+        result = f'Loaded {len(table_df)} rows FROM {table_name}.'
         print(table_df.info())
-        return table_df
-
+        return result

tutorial/read.py renamed to pandas_sqlalchemy_tutorial/read.py

@@ -1,5 +1,5 @@
 [tool.poetry]
-name = "Pandas SQLAlchemy Tutorial"
+name = "pandas_sqlalchemy_tutorial"
 version = "0.1.0"
 description = "Load or insert data into a SQL database using Pandas DataFrames."
 authors = ["Todd Birchard <toddbirchard@gmail.com>"]
@@ -18,7 +18,7 @@ keywords = ["Python",
 [tool.poetry.dependencies]
 python = "^3.8"
 pandas = "*"
-sqllchemy = "*"
+sqlalchemy = "*"
 pymysql = "*"
 python-dotenv = "*"
 

poetry.lock

@@ -1,8 +1,17 @@
+attrs==19.3.0
+mock==4.0.2
+more-itertools==8.3.0
 numpy==1.18.5
+packaging==20.4
 pandas==1.0.4
+pluggy==0.13.1
+py==1.8.1
 PyMySQL==0.9.3
+pyparsing==2.4.7
+pytest==5.4.3
 python-dateutil==2.8.1
 python-dotenv==0.13.0
 pytz==2020.1
 six==1.15.0
 SQLAlchemy==1.3.17
+wcwidth==0.2.4