src/helpers/database_handler.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
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()