init

Author: Iurii

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -0,0 +1,21 @@
+{
+  "install": "pip3 install --user -r requirements.txt",
+  "start": "uvicorn main:app",
+  "watch": {
+    "ignore": [
+      "\\.pyc$"
+    ],
+    "install": {
+      "include": [
+        "^requirements\\.txt$"
+      ]
+    },
+    "restart": {
+      "include": [
+        "\\.py$",
+        "^start\\.sh"
+      ]
+    },
+    "throttle": 1000
+  }
+}

glitch.json

@@ -0,0 +1,37 @@
+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from pymongo import MongoClient
+import json
+
+import os
+from dotenv import load_dotenv
+load_dotenv('.env')
+
+from src.models import load_data
+from src.bot import Bot, process_message
+
+# === GLOBAL OBJECTS ===
+client = MongoClient(os.environ.get('MONGODB_URI'))
+db = client.streak
+bot = Bot(token=os.environ['TELEGRAM_TOKEN'])
+templates = Jinja2Templates(directory="templates")
+app = FastAPI()
+app.mount("/static", StaticFiles(directory="static", html=True), name="static")
+
+
+@app.get("/", response_class=HTMLResponse)
+async def read_index(request: Request):
+  return templates.TemplateResponse(
+    "index.html", {
+      "participants": load_data(db.submissions.find()),
+      "request": request,
+    }
+  )
+
+
+@app.post("/hook")
+async def hook(request: Request):
+  process_message(bot, db, json.loads(await request.body()))
+  return "OK"

main.py

@@ -0,0 +1,5 @@
+uvicorn
+fastapi
+jinja2
+pymongo
+python-dotenv

requirements.txt

@@ -0,0 +1 @@
+__init__.py

src/__init__.py

@@ -0,0 +1,68 @@
+from datetime import datetime
+from pydantic import BaseModel
+import requests
+from src.common import submission_link
+
+
+LEETCODE_DAILY_GROUP_ID = '-1002270137956'
+
+
+class Bot(BaseModel):
+  token: str
+  
+  def api_url(self):
+    return f"https://api.telegram.org/bot{self.token}/"
+
+  def send_message(self, chat_id, message, useV2=False):
+      if chat_id is None or message is None:
+          return False
+
+      params = {
+          'chat_id': chat_id,
+          'text': message,
+          'disable_web_page_preview': True,
+      }
+      if useV2:
+        params['parse_mode'] = 'MarkdownV2'
+      res = requests.post(self.api_url() + "sendMessage", data=params).json()
+      print(res)
+
+      if res is None or 'result' not in res or 'message_id' not in res['result']:
+          return None
+      return res['result']['message_id']
+    
+    
+def process_message(bot, db, data):
+  try:
+    message = data['message']
+    chat_id = message['chat']['id']
+    text = message['text']
+    today = datetime.today().strftime('%Y-%m-%d')
+    
+    username = message['from']['username'] if 'username' in message['from'] else '😊'
+    first_name = message['from']['first_name'] if 'first_name' in message['from'] else '😊'
+    name = f'{first_name} ({username})'
+
+    if text.startswith('/start') or text.startswith('/help'):
+      bot.send_message(chat_id, "Please, send me a submission id for today's daily challenge")
+    elif text.isnumeric():
+      submission_id = text
+    
+    if submission_id.isnumeric():
+      print(f"Will update submissions for {username} on {today}: {text}")
+      result = db.submissions.update_one(
+        {'username': name, 'date': today, 'chat_id': chat_id},
+        {'$set': {'text': text}},
+        True,
+      )
+
+      print(username, today, text, result)
+      bot.send_message(chat_id, f"Updated submission for {today}: {submission_link(text)}")
+      bot.send_message(LEETCODE_DAILY_GROUP_ID, f"New submission from {name}: {submission_link(text)}")
+    else:
+      bot.send_message(chat_id, "Incorrect input! please send me a submission id")
+      raise ValueError(f"Incorrect link {data}")
+
+  except Exception as e:
+    print("Oops!")
+    print(e)

src/bot.py

@@ -0,0 +1,2 @@
+def submission_link(submission_id):
+  return f'https://leetcode.com/submissions/detail/{submission_id}'

src/common.py

@@ -0,0 +1,112 @@
+from datetime import datetime, timedelta
+from pydantic import BaseModel
+from typing import List
+from collections import defaultdict
+from src.common import submission_link
+from random import randint
+
+
+
+START_DATE = datetime(2024, 9, 16)
+
+
+class DisplaySubmission(BaseModel):
+  level: int
+  url: str
+  text: str
+  
+  
+class Participant(BaseModel):
+  name: str
+  reputation: int
+  submissions: List[DisplaySubmission]
+
+  
+def load_dummy_data():
+  return [
+    Participant(
+      name="Bob Marley",
+      reputation=55,
+      submissions=[
+        # week 1
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+
+        # week 2
+        DisplaySubmission(level=2, text="Submission", url=""),
+        DisplaySubmission(level=2, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=5, text="Submission", url=""),
+        DisplaySubmission(level=3, text="Submission", url=""),
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+
+      ],
+    ).dict(),
+
+    Participant(
+      name="John McClane",
+      reputation=45,
+      submissions=[
+        # week 1
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+
+        # week 2
+        DisplaySubmission(level=2, text="Submission", url=""),
+        DisplaySubmission(level=2, text="Submission", url=""),
+        DisplaySubmission(level=1, text="Submission", url=""),
+        DisplaySubmission(level=5, text="Submission", url=""),
+        DisplaySubmission(level=3, text="Submission", url=""),
+        DisplaySubmission(level=4, text="Submission", url=""),
+        DisplaySubmission(level=0, text="Submission", url=""),
+        
+      ],
+    ).dict(),
+  ]
+  
+  
+def load_data_from_cursor(cursor):
+  data = defaultdict(lambda: defaultdict(str))
+  users = defaultdict(str)
+  for record in cursor:
+    k = record['chat_id'] if 'chat_id' in record else record['username']
+    if 'username' in record:
+      users[k] = record['username']
+    data[k][record['date']] = record['text']
+    
+  
+  start_date = START_DATE
+  end_date = datetime.today()
+
+  result = []
+  days_of_week = ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"]
+  for k, p in data.items():
+    current_date = start_date
+    submissions = []
+    while current_date <= end_date:
+        d = current_date.strftime("%Y-%m-%d")
+        display_date = days_of_week[current_date.weekday()] + ", " + d
+        if d in p:
+          submissions.append(DisplaySubmission(level=4, text=display_date, url=submission_link(p[d])))
+        else:
+          submissions.append(DisplaySubmission(level=0, text=display_date, url=''))
+        current_date += timedelta(days=1)
+
+    result.append(Participant(name=users[k].split()[0], reputation=len(p), submissions=submissions).dict())
+  
+  result.sort(key=lambda p: (p['reputation'], randint(0, 100)), reverse=True)
+  return result
+
+def load_data(cursor=None, dummy=False):
+  return load_dummy_data() if dummy else load_data_from_cursor(cursor)