test hub merge (added roles up to 40)

app.py

@@ -100,6 +100,7 @@ def update_google_sheet():
         name = "levelbot"
         #worksheet = gspread_bot.open(name).sheet1
         global_df['discord_user_id'] = global_df['discord_user_id'].astype(str)
+        global_df['discord_exp'] = global_df['discord_exp'].astype(str)
         # could just do this / format
         set_with_dataframe(worksheet, global_df)
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -127,7 +128,7 @@ processed_users = set()
 
 async def periodic_api_test(): # needs rewrite, can do same thing and interact with csv -> push to google sheets
     try:
-        # this should be gspread type function (do not mix with discord at all)
+        # this should be gspread type function (do not mix with discord at all
         global worksheet2_df
         # get 
         for index, user in enumerate(worksheet2_df['hf_user_name']):
@@ -181,6 +182,7 @@ async def add_exp(member_id):
         lvl8 = guild.get_role(1171861968597024868)
         lvl9 = guild.get_role(1171862009982242836)
         lvl10 = guild.get_role(1164188093713223721)
+        
         lvl11 = guild.get_role(1171524944354607104)
         lvl12 = guild.get_role(1171524990257082458)
         lvl13 = guild.get_role(1171525021928263791)
@@ -202,11 +204,23 @@ async def add_exp(member_id):
         lvl28 = guild.get_role(1196055908267081849)
         lvl29 = guild.get_role(1196055970804150352)
         lvl30 = guild.get_role(1196056027720847380)
+
+        lvl31 = guild.get_role(1206542603261186078)
+        lvl32 = guild.get_role(1206542673549205514)
+        lvl33 = guild.get_role(1206542690939048007)
+        lvl34 = guild.get_role(1206542707862806568)
+        lvl35 = guild.get_role(1206542723633512468)
+        lvl36 = guild.get_role(1206542738728681485)
+        lvl37 = guild.get_role(1206542754625101866)
+        lvl38 = guild.get_role(1206542771314364416)
+        lvl39 = guild.get_role(1206542785973321758)
+        lvl40 = guild.get_role(1206542802155208725)        
         
         lvls = {
             1: lvl1, 2: lvl2, 3: lvl3, 4: lvl4, 5: lvl5, 6: lvl6, 7: lvl7, 8: lvl8, 9: lvl9, 10: lvl10,
             11: lvl11, 12: lvl12, 13: lvl13, 14: lvl14, 15: lvl15, 16: lvl16, 17: lvl17, 18: lvl18, 19: lvl19, 20: lvl20,
             21: lvl21, 22: lvl22, 23: lvl23, 24: lvl24, 25: lvl25, 26: lvl26, 27: lvl27, 28: lvl28, 29: lvl29, 30: lvl30,
+            31: lvl31, 32: lvl32, 33: lvl33, 34: lvl34, 35: lvl35, 36: lvl36, 37: lvl37, 38: lvl38, 39: lvl39, 40: lvl40,
         }              
 
         # xp = int64
@@ -219,6 +233,7 @@ async def add_exp(member_id):
 
         print("test1")
         # discord_user_id column
+        # iterate over items of first column (discord_user_id)
         for index, cell_value in global_df.iloc[:, 0].items():
             # print("test2")
             # remove L (read, so no need to replace)
@@ -239,35 +254,73 @@ async def add_exp(member_id):
             
                     # increment the old experience value (better not to replace outright)
                     old_xp = global_df.loc[index, 'discord_exp']
+
+                    # add to discord exp (this is input)
+                    # add discord exp and hub exp to total_exp
+                    # set level based on total_exp (this is output)
+
                     # remove L (write, so we replace)
                     old_xp = str(old_xp)
                     if old_xp.startswith("L") and old_xp.endswith("L"):  
                         old_xp = old_xp[1:-1]
-                    # str -> int temporarily
+
+                    # set old level; use this for more accurate logging and jumping multiple levels at once (for example, verifying)
+                    old_level = calculate_level(int(old_xp))
+                        
+                    # str -> int temporarily for adding
                     new_xp = int(old_xp) + XP_PER_MESSAGE
-                    # check + convert back to string + google sheet proofing
-                    current_level = calculate_level(new_xp) 
+
+                    # ----------------------------------------------------
+                    total_exp = global_df.loc[index, 'total_exp']
+                    hub_xp = global_df.loc[index, 'hub_exp']
+                    
+                    total_exp = str(total_exp)
+                    hub_xp = str(hub_xp)
+                    
+                    if total_exp.startswith("L") and total_exp.endswith("L"):  
+                        total_exp = total_exp[1:-1]
+                    if hub_xp.startswith("L") and hub_xp.endswith("L"):  
+                        hub_xp = hub_xp[1:-1]   
+
+                    old_total_xp = int(total_exp)
+
+                    # check if hub exp not empty
+                    if hub_xp.strip():  
+                        total_exp = int(new_xp) + int(hub_xp)
+                    else:
+                        total_exp = int(new_xp)                    
+                    # ----------------------------------------------------
+                    
+                    # total v
+                    current_level = calculate_level(total_exp) 
+
+                    # convert back to string + google sheet proofing
                     new_xp = str(new_xp)
                     if not new_xp.startswith("L") and not new_xp.endswith("L"): 
                         new_xp = f"L" + str(new_xp) + f"L"
+                    global_df.loc[index, 'discord_exp'] = new_xp # do not change column name
 
+                    # after
+                    total_exp = str(total_exp)
+                    if not total_exp.startswith("L") and not total_exp.endswith("L"): 
+                        total_exp = f"L" + str(total_exp) + f"L" 
+                    
                     # add back to dataframe in memory after checking redundantly; 
-                    if new_xp.startswith("L") and new_xp.endswith("L"): 
+                    if total_exp.startswith("L") and total_exp.endswith("L"): 
                         print("test5")
-                        global_df.loc[index, 'discord_exp'] = new_xp # do not change column name
-                        print(f"Record for {member} updated from {old_xp} to {global_df.loc[index, 'discord_exp']} (+{XP_PER_MESSAGE}) ")
+                        global_df.loc[index, 'total_exp'] = total_exp # do not change column name 
+                        print(f"Record for {member} updated from {old_total_xp} to {global_df.loc[index, 'total_exp']} (+{XP_PER_MESSAGE}) ")
                   
                         # level up
-                        
                         verified_role = guild.get_role(900063512829755413)
                         
                         print(f"Current_level for {member}: {current_level}")
-                        if current_level >= 2 and current_level <=30:
+                        if current_level >= 2 and current_level <=40:
                             print("test6")
                             current_role = lvls[current_level]
-                            if current_role not in member.roles: # if we leveled up
+                            if current_role not in member.roles: # if we need to level up / update role
         
-                                # finding leaderboard rank + excluding huggingfolks 
+                                # finding leaderboard rank + excluding huggingfolks (still need exclusion)
                                 try:
                                     # cell_value = L1234567890L
                                     # make copy
@@ -275,12 +328,12 @@ async def add_exp(member_id):
                                     # do calculations
                                     copy_df = global_df.copy()
                                     copy_df['discord_user_id'] = copy_df['discord_user_id'].str.strip('L').astype(str)
-                                    copy_df['discord_exp'] = copy_df['discord_exp'].str.strip('L').astype(int)
+                                    copy_df['total_exp'] = copy_df['total_exp'].str.strip('L').astype(int)
                                     row = copy_df[copy_df['discord_user_id'] == str(member_id)]
                                     print(f"Row found for {member_id}!")
-                                    target_exp = row['discord_exp'].values[0]
-                                    rank = (copy_df['discord_exp'] > target_exp).sum() + 1
-                                    print(f"The rank for discord_id {member_id} based on discord_exp is: {rank}")
+                                    target_exp = row['total_exp'].values[0]
+                                    rank = (copy_df['total_exp'] > target_exp).sum() + 1
+                                    print(f"The rank for discord_id {member_id} based on total_exp is: {rank}")
                                 except Exception as e:
                                     print(f"Discord ID {member_id} not found in the DataFrame.") 
                                     rank = "🤗"
@@ -291,15 +344,15 @@ async def add_exp(member_id):
                                     if verified_role not in member.roles:     
 
 
-                                        # XP has L; remove, calculate
-                                        new_xp = new_xp[1:-1]
-                                        new_xp = int(new_xp)
+                                        # L12345L -> `12345` -> 12345
+                                        total_exp = total_exp[1:-1]
+                                        total_exp = int(total_exp)
                                         
-                                        if new_xp % 10 == 0: # staggers messages so we don't send one every time exp is earned
+                                        if total_exp % 10 == 0: # staggers messages so we don't send one every time exp is earned
                                             # claim exp (-30 for level 3, but +100 as bonus exp. This scales infinitely until the member verifies,
                                             # so they can continue earning exp, it just won't translate to levels and the leaderboard.
                                             # This way they can claim at any time and get a big boost in levels!
-                                            claim_exp = new_xp + 70
+                                            claim_exp = total_exp + 70
                                                                  
                                             # send embed 
                                             embed = Embed(color=Color.red())
@@ -317,19 +370,20 @@ async def add_exp(member_id):
                                             embed.set_image(url='https://cdn.discordapp.com/attachments/1150399343912833024/1205537451242688573/download_1.png?ex=65d8bb3e&is=65c6463e&hm=042fe7dd3521887db0bd48eeb846de1cc7c75194f9e95215c23512ff61ea3475&')
                                             
                                             lunar = bot.get_user(811235357663297546)
-                                            await member.send(embed=embed)  
+                                            #await member.send(embed=embed)  
+                                            await lunar.send(embed=embed)  
                                             print(f"Sent verification cap embed to {member}")
                                             return
                                     
                                     
                                 # increment the old level value (better to replace outright)
-                                # only increment level column if you are lvl2 or 3+ with verified role
+                                # only increment level column if you are lvl2 or 3+ with verified role (this may make some members not appear)
                                 global_df.loc[index, 'discord_level'] = current_level # do not change column name  
                                 
                                 await member.add_roles(current_role)
                                 print(f"Level Up! Gave {member} {current_role}")
-                                await member.remove_roles(lvls[current_level-1])
-                                print(f"Removed {lvls[current_level-1]} from {member}")  
+                                await member.remove_roles(lvls[old_level])
+                                print(f"Removed {lvls[old_level]} from {member}")  
                                 #print(f"{member} Level up! {current_level-1} -> {current_level}!")
                                 if current_role in member.roles: # needs update; reference exp reward for verification
                                     #await member.send(f"Level up! {current_level-1} -> {current_level}!")  
@@ -339,7 +393,7 @@ async def add_exp(member_id):
                                     # send embed
                                     embed = Embed(color=Color.blue())
                                     embed.set_author(name=f"{member}", icon_url=member.avatar.url if member.avatar else bot.user.avatar.url)
-                                    embed.title = f"Level Up! `{current_level-1}` -> `{current_level}`"
+                                    embed.title = f"Level Up! `{old_level-1}` -> `{current_level}`"
                                     msg = f'🤗 Congrats {member}! You just leveled up in the Hugging Face Discord server'
                                     embed.description = f"{msg}."
                                     embed.add_field(name="Leaderboard Ranking:", value=f"👑 **{rank}**\n\nhttps://discord.com/channels/879548962464493619/1197143964994773023", inline=True)
@@ -356,7 +410,8 @@ async def add_exp(member_id):
                                     embed.add_field(name="Verify Here:", value=verification_link, inline=True)
         
                                     lunar = bot.get_user(811235357663297546)
-                                    await member.send(embed=embed)                           
+                                    #await member.send(embed=embed)
+                                    await lunar.send(embed=embed)  
                                     print(f"Sent levelup embed to {member}")
                                     #You can verify your account to earn 100 points! To verify, do A.                            
 
@@ -431,17 +486,16 @@ async def remove_huggingfolks():
         # make a copy while discord id column still exists -> use for rank in discord embeds
         community_global_df_with_id = community_global_df_with_id.copy()
 
-
-
-
-
         # drop first column (discord id -> this is so we can display the important stuff in the leaderboard)
         community_global_df.drop(community_global_df.columns[0], axis=1, inplace=True)
-        community_global_df['discord_exp'] = community_global_df['discord_exp'].str.strip('L').astype(int)
-        community_global_df['discord_exp'] = pd.to_numeric(community_global_df['discord_exp'])
-        community_global_df = community_global_df.nlargest(len(community_global_df), 'discord_exp')        
+        community_global_df.drop(community_global_df.columns[1], axis=1, inplace=True)
+        community_global_df.drop(community_global_df.columns[2], axis=1, inplace=True)
+        community_global_df.drop(community_global_df.columns[2], axis=1, inplace=True)
+        community_global_df['total_exp'] = community_global_df['total_exp'].str.strip('L').astype(int)
+        community_global_df['total_exp'] = pd.to_numeric(community_global_df['total_exp'])
+        community_global_df = community_global_df.nlargest(len(community_global_df), 'total_exp')        
         
-        top_30_exp = community_global_df.nlargest(30, 'discord_exp')
+        top_30_exp = community_global_df.nlargest(30, 'total_exp')
         
         top_30_exp['D'] = ['🥇','🥈','🥉','','','','','','','','','','','','','','','','','','','','','','','','','','','']
         top_30_rows = top_30_exp.values.tolist()