diff options
| author | Bobby <[email protected]> | 2022-12-02 17:02:31 -0500 |
|---|---|---|
| committer | GitHub <[email protected]> | 2022-12-02 17:02:31 -0500 |
| commit | af2ada7022d75411f2d74a4a4d2c95dfe3eb2e3a (patch) | |
| tree | 327819303a94662dfcfd921864d2e1134fffd93e /src/helpers/database_handler.py | |
| parent | 0eb9ac45dcf195e65ea4941229c7e32f17b46d87 (diff) | |
| parent | c4f746246a66ed644f8cd123f1dd0081abcd1a55 (diff) | |
| download | RedditEngagementPrediction-af2ada7022d75411f2d74a4a4d2c95dfe3eb2e3a.tar.xz RedditEngagementPrediction-af2ada7022d75411f2d74a4a4d2c95dfe3eb2e3a.zip | |
Merge pull request #8 from luciferreeves/main
Tkinter Application Added
Diffstat (limited to 'src/helpers/database_handler.py')
| -rw-r--r-- | src/helpers/database_handler.py | 15 |
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 |