From d73aa25bd619edc71c9b5363623ca43973a7f008 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Wed, 24 Nov 2021 21:38:03 -0500 Subject: [PATCH] [M] Move Reporter to Utils --- src/process/twitter_visualization.py | 34 ---------------------------- src/utils.py | 28 +++++++++++++++++++++++ 2 files changed, 28 insertions(+), 34 deletions(-) diff --git a/src/process/twitter_visualization.py b/src/process/twitter_visualization.py index 1cb5f20..dd4b5e4 100644 --- a/src/process/twitter_visualization.py +++ b/src/process/twitter_visualization.py @@ -12,34 +12,6 @@ from constants import REPORT_DIR from process.twitter_process import * -class Reporter: - """ - Report file creator - """ - report: str - file: str - - def __init__(self, file: str) -> None: - self.report = '' - self.file = file - - def print(self, line: str = '', arg: Any = None) -> None: - """ - Add a line to the report - - :param line: Line content - :param arg: Additional argument - :return: None - """ - self.report += line - if arg is not None: - self.report += str(arg) - self.report += '\n' - - def save(self) -> None: - write(self.file, self.report) - - @dataclass() class UserFloat: """ @@ -245,12 +217,6 @@ def view_covid_tweets_date(tweets: list[Posting]): if __name__ == '__main__': - - view_covid_tweets_freq([u.username for u in samples.most_popular], '500-pop') - # view_covid_tweets_freq(sample.random, '500-rand') - # view_covid_tweets_pop(sample.most_popular, '500-pop') - # view_covid_tweets_pop(sample.random, '500-rand') - # samples = load_user_sample() # combine_tweets_for_sample([u.username for u in samples.most_popular], '500-pop') # combine_tweets_for_sample([u.username for u in samples.random], '500-rand') diff --git a/src/utils.py b/src/utils.py index b4e316b..70b00dd 100644 --- a/src/utils.py +++ b/src/utils.py @@ -99,6 +99,34 @@ def read(file: str) -> str: return f.read() +class Reporter: + """ + Report file creator + """ + report: str + file: str + + def __init__(self, file: str) -> None: + self.report = '' + self.file = file + + def print(self, line: str = '', arg: Any = None) -> None: + """ + Add a line to the report + + :param line: Line content + :param arg: Additional argument + :return: None + """ + self.report += line + if arg is not None: + self.report += str(arg) + self.report += '\n' + + def save(self) -> None: + write(self.file, self.report) + + def remove_outliers(points: list[float], z_threshold: float = 3.5) -> list[float]: """ Create list with outliers removed for graphing