[M] Move Reporter to Utils
This commit is contained in:
@@ -12,34 +12,6 @@ from constants import REPORT_DIR
|
|||||||
from process.twitter_process import *
|
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()
|
@dataclass()
|
||||||
class UserFloat:
|
class UserFloat:
|
||||||
"""
|
"""
|
||||||
@@ -245,12 +217,6 @@ def view_covid_tweets_date(tweets: list[Posting]):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
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()
|
# 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.most_popular], '500-pop')
|
||||||
# combine_tweets_for_sample([u.username for u in samples.random], '500-rand')
|
# combine_tweets_for_sample([u.username for u in samples.random], '500-rand')
|
||||||
|
|||||||
@@ -99,6 +99,34 @@ def read(file: str) -> str:
|
|||||||
return f.read()
|
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]:
|
def remove_outliers(points: list[float], z_threshold: float = 3.5) -> list[float]:
|
||||||
"""
|
"""
|
||||||
Create list with outliers removed for graphing
|
Create list with outliers removed for graphing
|
||||||
|
|||||||
Reference in New Issue
Block a user