From d51a4e636784233b6e8a0e0e862acbea4863315a Mon Sep 17 00:00:00 2001
From: Rykov7 <rykov7@protonmail.ch>
Date: Fri, 17 Feb 2023 12:28:11 +0300
Subject: [PATCH] add rollback

---
 Stan.txt      | 8 +++++++-
 app/config.py | 5 +++--
 app/report.py | 6 +++++-
 3 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/Stan.txt b/Stan.txt
index daed886..f1f3131 100644
--- a/Stan.txt
+++ b/Stan.txt
@@ -152,7 +152,6 @@ I'm not hear for jokes, strictly business.
 К субботе готов
 Ну можно гит клон попробовать
 подумай разберись почини, этож опенсорс
-У меня что в нике написано, Ставитель скрипов автопродаж?
 пусть тренируются, песдюки
 Ну так вы мне проекции показываете
 и чо лям, когда есть пакет
@@ -357,3 +356,10 @@ facepalm.jpg
 🦔
 наш человек че там
 хренасси год начинается, в чате питон обсуждают
+Кто ищет - тот всегда найдёт
+ниче ниче в 12 карета превратится в тыкву и имплицитность в инкапсуляцию
+какое образование, такой и скрипт
+сие зело меня печалит
+у каждого был такой знакомый...
+Гвидо наносит ответный удар
+If you are new to anything, follow tutorials
diff --git a/app/config.py b/app/config.py
index 9bec3d9..6890123 100644
--- a/app/config.py
+++ b/app/config.py
@@ -29,8 +29,9 @@ ALLOWED_WORDS = ['paste', 'nekobin', 'github', 'google', 'nometa', 'python', 'dj
                  'stackoverflow', 'habr', 'medium', 'youtu', 'stepik', 'telegra', '#rtfm', 'support',
                  'jetbrains', ]
 
-WHITEUN = set(os.environ.get('whitelist', '<<<ERR_USRS').split(','))
-WHITEIDS = {int(i) for i in os.environ.get('whiteids', '<<<ERR_IDS').split(',')}
+WHITEUN = set(os.environ.get('whitelist').split(','))
+WHITEIDS = {int(i) for i in os.environ.get('whiteids').split(',')}
+ROLLBACK = {int(i) for i in os.environ.get('rollback').split(',')}
 
 RUS = """ёйцукенгшщзхъфывапролджэячсмитьбю.Ё!"№;%:?ЙЦУКЕНГШЩЗХЪ/ФЫВАПРОЛДЖЭЯЧСМИТЬБЮ,"""
 ENG = """`qwertyuiop[]asdfghjkl;'zxcvbnm,./~!@#$%^&QWERTYUIOP{}|ASDFGHJKL:"ZXCVBNM<>?"""
diff --git a/app/report.py b/app/report.py
index 70e2a1c..2e5d39b 100644
--- a/app/report.py
+++ b/app/report.py
@@ -1,6 +1,6 @@
 import shelve
 import os
-from .config import DATA
+from .config import DATA, ROLLBACK
 
 
 def create_report_text(chat_id):
@@ -9,6 +9,10 @@ def create_report_text(chat_id):
     flooders = []
     if os.path.exists(f'{DATA}{chat_id}.db'):
         with shelve.open(f'{DATA}{chat_id}') as s:
+            for user_id in ROLLBACK:
+                if user_id in s['Messages'] and s['Messages'][user_id]['Count'] > 10:
+                    s['Messages'][user_id]['Count'] //= 2
+
             for n in range(min(3, len(s['Messages']))):
                 top_user = s['Messages'][
                     sorted(s['Messages'], key=lambda a: s['Messages'][a]['Count'], reverse=True)[n]]