[M] Move Reporter to Utils

This commit is contained in:
Hykilpikonna
2021-11-24 21:38:03 -05:00
parent cc90af631d
commit d73aa25bd6
2 changed files with 28 additions and 34 deletions
-34
View File
@@ -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')
+28
View File
@@ -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