Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: making heatmaps less intensive! #104

Merged
merged 3 commits into from
Jul 10, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 19 additions & 15 deletions discord_analyzer/analyzer/analyzer_heatmaps.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,10 +100,6 @@ def analysis_heatmap(self, guildId: str, from_start: bool = False):
guildId=guildId,
)

account_list = get_userids(
db_mongo_client=self.DB_connections.mongoOps.mongo_db_access.db_mongo_client,
guildId=guildId,
)
while last_date.date() < datetime.now().date():
entries = rawinfo_c.get_day_entries(last_date, "ANALYZER HEATMAPS: ")
if len(entries) == 0:
Expand All @@ -112,36 +108,44 @@ def analysis_heatmap(self, guildId: str, from_start: bool = False):
continue

prepared_list = []
account_list = []

for entry in entries:
if "replied_user" not in entry:
reply = ""
else:
reply = entry["replied_user"]
if reply not in account_list and reply not in bot_ids:
account_list.append(reply)

# eliminating bots
if entry["author"] not in bot_ids:
author = entry["author"]
mentioned_users = entry["user_mentions"]
if author not in bot_ids:
prepared_list.append(
{
# .strftime('%Y-%m-%d %H:%M'),
"datetime": entry["createdDate"],
"channel": entry["channelId"],
"author": entry["author"],
"author": author,
"replied_user": reply,
"user_mentions": entry["user_mentions"],
"user_mentions": mentioned_users,
"reactions": entry["reactions"],
"threadId": entry["threadId"],
"mess_type": entry["type"],
}
)
if (
entry["author"] not in account_list
and entry["author"] not in bot_ids
):
account_list.append(entry["author"])

if entry["user_mentions"] is not None:
for account in entry["user_mentions"]:
if author not in account_list:
account_list.append(author)

# Adding the interacting users
for users_reacting in entry["reactions"]:
for user in users_reacting.split(",")[:-1]:
if user not in account_list:
account_list.append(user)

if mentioned_users is not None:
for account in mentioned_users:
if account not in account_list and account not in bot_ids:
account_list.append(account)

Expand Down
Loading