hfapi

app.py

@@ -36,7 +36,7 @@ with open(file_path, 'w') as json_file:
     json.dump(service_account, json_file)
 gspread_bot = gspread.service_account(filename='service_account.json')
 #worksheet = gspread_bot.open("levelbot").sheet1
-#worksheet2 = gspread_bot.open("hf_discord_verified_users_test").sheet1
+worksheet2 = gspread_bot.open("hf_discord_verified_users_test").sheet1
 """"""
 bot_ids = [1136614989411655780, 1166392942387265536, 1158038249835610123, 1130774761031610388, 1155489509518098565, 1155169841276260546, 1152238037355474964, 1154395078735953930]
 """"""
@@ -60,6 +60,43 @@ def calculate_xp(level):
 
 
 
+processed_users = set()
+
+async def periodic_api_test():
+    try:
+        column_values_3 = worksheet2.col_values(3)
+        column_values_8 = worksheet2.col_values(8)
+    
+        for i, user in enumerate(column_values_3):
+            if user in processed_users:
+                continue
+    
+            url = f"https://huggingface.co/api/users/{user}/overview"
+            response = requests.get(url)
+    
+            if response.status_code == 200:
+                data = response.json()
+    
+                likes = data["numLikes"]
+                models = data["numModels"]
+                datasets = data["numDatasets"]
+                spaces = data["numSpaces"]
+                discussions = data["numDiscussions"]
+                papers = data["numPapers"]
+                upvotes = data["numUpvotes"]
+    
+                worksheet2.update(values=[[likes, models, datasets, spaces, discussions, papers, upvotes]],
+                                  range_name=f'G{i+1}:M{i+1}')
+    
+                processed_users.add(user)
+            else:
+                print(f"Failed to retrieve data for user {user}. Status code: {response.status_code}")
+            
+    except Exception as e:
+        print(f"periodic_api_test Error: {e}")   
+
+
+
 
 """
 async def search_csv(file_path, column_index, search_value):
@@ -215,7 +252,7 @@ async def on_message(message):
         if message.author.id not in bot_ids:
             print(f"adding exp from message {message.author}")
             await add_exp(message.author.id)
-            #await periodic_api_test() # run this ~1 per day otherwise will lead to issues
+            await periodic_api_test() # run this ~1 per day otherwise will lead to issues
         await bot.process_commands(message)
     except Exception as e:
         print(f"on_message Error: {e}")