azalea/CSC110-Project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix name shadowing

Browse Source
This commit is contained in:
Hykilpikonna
2021-11-22 10:49:37 -05:00
parent 29270cff70
commit 8049a3604d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/raw_collect/twitter.py
+4 -4
View File
@@ -249,17 +249,17 @@ def download_users_execute(api: API, n: float, base_dir: str, rate_limit: int,
# debug(f'- Downloaded {user.screen_name}') # debug(f'- Downloaded {user.screen_name}')
# Get users and their popularity that we haven't downloaded # Get users and their popularity that we haven't downloaded
screen_names = [(user.screen_name, user.followers_count) for user in friends screen_names = [(u.screen_name, u.followers_count) for u in friends
if user.screen_name not in done_set and not user.protected] if u.screen_name not in done_set and not u.protected]
# Sort by followers count, from least popular to most popular # Sort by followers count, from least popular to most popular
screen_names.sort(key=lambda x: x[1]) screen_names.sort(key=lambda x: x[1])
# Add 3 random users to the next set # Add 3 random users to the next set
if len(screen_names) > 3: if len(screen_names) > 3:
samples = {n[0] for n in random.sample(screen_names, 3)} samples = {u[0] for u in random.sample(screen_names, 3)}
else: else:
samples = {n[0] for n in screen_names} samples = {u[0] for u in screen_names}
# Add 3 most popular users that we haven't downloaded to the next set # Add 3 most popular users that we haven't downloaded to the next set
while len(screen_names) > 0 and len(samples) < 6: while len(screen_names) > 0 and len(samples) < 6: