From 426a523a477d3a8235cfb53997dd0701208f5bd1 Mon Sep 17 00:00:00 2001
From: emmatveev <emmtvv@gmail.com>
Date: Sun, 31 Mar 2024 10:03:04 +0300
Subject: [PATCH] ebal

---
 main.py  | 9 +++++++++
 mongo.py | 6 ++++++
 2 files changed, 15 insertions(+)

diff --git a/main.py b/main.py
index fcd50b8..0b3ca90 100644
--- a/main.py
+++ b/main.py
@@ -62,6 +62,15 @@ def show_rating(message: Message):
     bot.send_message(message.chat.id, text)
 
 
+@bot.message_handler(commands=['point'])
+def show_rating(message: Message):
+    if not message.reply_to_message:
+        bot.reply_to(message, 'Чтобы начислить Ебаллы, нужно прописать команду ответом на чье-то сообщение')
+    username = message.reply_to_message.from_user.username
+    mongo.inc_points(username, message.chat.id)
+    bot.reply_to(message.reply_to_message, 'Тебе начислили Ебалл!')
+
+
 @bot.message_handler()
 def do_action(message: Message):
     if message.reply_to_message:
diff --git a/mongo.py b/mongo.py
index 619440f..5b2ad84 100644
--- a/mongo.py
+++ b/mongo.py
@@ -35,5 +35,11 @@ class Mongo:
         else:
             self.counter_collection.insert_one({"chat_id": chat_id, "username": username, "count": 1})
 
+    def inc_points(self, username, chat_id):
+        if self.counter_collection.find_one({"chat_id": chat_id, "username": username}):
+            self.counter_collection.update_one({"chat_id": chat_id, "username": username}, {"$inc": {"points": 1}})
+        else:
+            self.counter_collection.insert_one({"chat_id": chat_id, "username": username, "points": 1})
+
 
 mongo = Mongo()