blob: bb82b4d8675c0662b9fe9cf84027c8678870101b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
import tkinter
import tkinter.messagebox
import pandas as pd
from helpers.database_handler import DatabaseHandler
from helpers.preprocessor import Preprocessor
from windows.data_fetcher import DataDownloader
from windows.plotviewer import PlotViewer
def fetch_data():
downloader = DataDownloader()
downloader.start()
return downloader.posts
if DatabaseHandler().read().empty:
# show a message box
response = tkinter.messagebox.askokcancel('No Data Found', 'No data found in database. Do you want to fetch data from Reddit?', icon='warning')
if response:
posts = fetch_data()
# ask if user wants to save the data
response = tkinter.messagebox.askokcancel('Save Data', 'Do you want to save the data for future use?', icon='warning')
posts = pd.DataFrame(posts)
posts = Preprocessor(posts).get_preprocessed_data()
if response:
DatabaseHandler().write(posts)
else:
posts = DatabaseHandler().read()
if __name__ == '__main__':
plot_viewer = PlotViewer(posts)
plot_viewer.mainloop()
|
