From 461f83918c11775ea39200a3d4579208c8f2b57a Mon Sep 17 00:00:00 2001
From: Hykilpikonna <me@hydev.org>
Date: Tue, 23 Nov 2021 11:25:21 -0500
Subject: [PATCH] [+] Load user sample

---
 src/process/twitter_process.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/src/process/twitter_process.py b/src/process/twitter_process.py
index ba6e3a6..8e0d38f 100644
--- a/src/process/twitter_process.py
+++ b/src/process/twitter_process.py
@@ -5,6 +5,7 @@ from datetime import datetime, time
 from typing import NamedTuple
 
 from utils import *
+from dataclasses import dataclass
 
 
 class UserPopularity(NamedTuple):
@@ -83,6 +84,12 @@ def get_user_popularity_ranking(user: str, user_dir: str = './data/twitter/user/
     return -1
 
 
+@dataclass()
+class Sample:
+    most_popular: list[UserPopularity]
+    random: list[UserPopularity]
+
+
 def select_user_sample(user_dir: str = './data/twitter/user/') -> None:
     """
     Select our sample of 500 most popular users and 500 random users who meet the criteria. The
@@ -113,7 +120,24 @@ def select_user_sample(user_dir: str = './data/twitter/user/') -> None:
 
     # Save
     write(f'{user_dir}/processed/sample.json',
-          json_stringify({'most_popular': most_popular, 'random': sample}))
+          json_stringify(Sample(most_popular, sample)))
+
+
+def load_user_sample(user_dir: str = './data/twitter/user/') -> Sample:
+    """
+    Load the selected sample
+
+    :param user_dir: Download directory for users
+    :return: None
+    """
+    user_dir = normalize_directory(user_dir)
+
+    j = json.loads(read(f'{user_dir}/processed/sample.json'))
+    return Sample([UserPopularity(*u) for u in j['most_popular']],
+                  [UserPopularity(*u) for u in j['random']])
+
+
+
 
 class Posting(NamedTuple):
     """