From 592176202e1159d9d85b21f50ddf17460ebede97 Mon Sep 17 00:00:00 2001 From: Bobby Date: Fri, 2 Dec 2022 17:01:06 -0500 Subject: Tkinter Application --- src/helpers/database_handler.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 src/helpers/database_handler.py (limited to 'src/helpers/database_handler.py') 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 -- cgit v1.2.3