Skip to content

Commit

Permalink
'Refactored by Sourcery'
Browse files Browse the repository at this point in the history
  • Loading branch information
SourceryAI committed Sep 15, 2021
1 parent fb7330e commit 52cbb03
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 18 deletions.
28 changes: 14 additions & 14 deletions bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,9 +134,7 @@ def handle_text(message):
admins = db.all_admins()
text = "All admins:\n\n"
for admin in admins:
if str(admin[0]) == 'none':
pass
else:
if str(admin[0]) != 'none':
text += "<b>@%s</b>\n" % (str(admin[0]))
bot.send_message(message.from_user.id, text, parse_mode='HTML')

Expand Down Expand Up @@ -200,17 +198,19 @@ def del_warning(message):

@bot.message_handler(commands=['allwarnings'])
def handle_text(message):
if message.chat.type == "private" and message.from_user.id in superadmins:
warnings = db.all_warnings()
text = "All warnings:\n\n"
for warning in warnings:
if str(warning[0]) == 'none':
pass
else:
text += "<b>@%s</b>: %s\n" % (str(warning[0]), str(warning[1]))
splitted_text = util.split_string(text, 3000)
for text in splitted_text:
bot.send_message(message.chat.id, text, parse_mode='HTML')
if (
message.chat.type != "private"
or message.from_user.id not in superadmins
):
return
warnings = db.all_warnings()
text = "All warnings:\n\n"
for warning in warnings:
if str(warning[0]) != 'none':
text += "<b>@%s</b>: %s\n" % (str(warning[0]), str(warning[1]))
splitted_text = util.split_string(text, 3000)
for text in splitted_text:
bot.send_message(message.chat.id, text, parse_mode='HTML')


# ============================================= #
Expand Down
6 changes: 2 additions & 4 deletions dbhelper.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,12 +108,10 @@ def del_warning(self, insta_username):

def all_warnings(self):
self.c.execute("SELECT insta_user, warnings FROM data")
warnings = self.c.fetchall()
return warnings
return self.c.fetchall()

def all_admins(self):
self.c.execute("SELECT insta_user FROM data WHERE admin=1")
admins = self.c.fetchall()
return admins
return self.c.fetchall()


0 comments on commit 52cbb03

Please sign in to comment.