Tkinter Application

1 files changed, 15 insertions, 0 deletions

diff --git a/src/helpers/database_handler.py b/src/helpers/database_handler.py
new file mode 100644
index 0000000..af86e6e
--- /dev/null
+++ b/src/helpers/database_handler.py
@@ -0,0 +1,15 @@
+import pandas as pd
+
+
+class DatabaseHandler:
+
+    def write(self, df):
+        # save the data to csv
+        df.to_csv('reddit.csv', index=False)
+
+    def read(self):
+        try:
+            df = pd.read_csv('reddit.csv')
+            return df
+        except FileNotFoundError:
+            return pd.DataFrame()
\ No newline at end of file