async def on_message(self, message):
if not message.content.startswith(
self.bot.command_prefix) and not message.author.bot:
guild = message.guild
author = message.author
channel = message.channel
if guild is not None:
with self.cursor_context() as cursor:
db_util.select("statistics_global").items(
"id",
"msg_count").where(id_server=guild.id,
id_user=author.id,
id_channel=channel.id).run(cursor)
row = cursor.fetchone()
with self.cursor_context(commit=True) as cursor:
if not row:
db_util.insert("statistics_global").items(
id_server=guild.id,
id_user=author.id,
id_channel=channel.id,
msg_count=1).run(cursor)
else:
row_id = row[0]
db_util.update("statistics_global").items(
msg_count=DBFunction("msg_count+1")).where(
id=row_id).run(cursor)
async def set(self, ctx, tz_name: str):
"""Set your current timezone"""
try:
local_tz = pytz.timezone(tz_name)
except pytz.exceptions.UnknownTimeZoneError:
await ctx.reply(f'The timezone "**{tz_name}**" does not exist')
return
with self.cursor_context(commit=True) as cursor:
cursor.execute(
*db_util.select("profiles").items("id").limit(1).where(
id_user=ctx.author.id).build)
row = cursor.fetchone()
with self.cursor_context(commit=True) as cursor:
if not row:
cursor.execute(*db_util.insert("profiles").items(
id_user=ctx.author.id, timezone=tz_name).build)
else:
cursor.execute(*db_util.update("profiles").items(
timezone=tz_name).where(id=row[0]).build)
utc = pytz.utc
time_now = datetime.datetime.utcnow()
time_now_utc = utc.localize(time_now)
time_now_localized = time_now_utc.astimezone(local_tz)
time_now_formatted = time_now_localized.strftime('%Y-%m-%d %H:%M:%S')
await ctx.reply(
f'Your timezone has been set to "**{tz_name}**", your local time is **{time_now_formatted}**'
)
async def process(self, ctx, timestamp: str = None):
"""Process every question with your :upvote: reaction on it, save it to the database and remove it"""
saved = []
with self.cursor_context() as cursor:
# Get the last stream ID
res = cursor.execute(*db_util.select("streams", "id").limit(1)
.where(id_server=ctx.guild.id).order(date="desc").build)
if not res:
await ctx.reply('No streams have been found !')
return
row = cursor.fetchone()
stream_id = row[0]
destination = self.bot.get_channel(self.config.bot.channel.ama_destination)
if destination is None:
await ctx.reply('There is no destination channel.')
return
async for msg in ctx.channel.history(limit=200):
to_save = False
for reaction in msg.reactions:
if reaction.emoji.name == 'upvote':
from_me = discord.utils.get(reaction.users().flatten(), id=ctx.author.id) is not None
to_save = from_me
break
if to_save:
question_details = msg.content.split('\n-----------------------\n')
if len(question_details) != 2:
await ctx.reply('question_details fail.')
continue
question_infos = question_details[0].split(' | ')
if len(question_infos) != 3:
await ctx.reply('question_infos fail.')
continue
saved.append(msg)
q_content = question_details[1]
q_author = question_infos[0].replace('From ', '')
q_date = datetime.strptime(question_infos[1].replace(' UTC', ''), '%c')
q_timestamp = '' if timestamp is None else timestamp
with self.cursor_context(commit=True) as cursor:
cursor.execute(*db_util.insert("questions")
.items(id_server=ctx.guild.id, id_stream=stream_id, author=q_author,
datetime=q_date.strftime('%Y-%m-%d %H:%M:%S'), question=q_content,
timestamp=q_timestamp))
if saved:
for msg in saved:
await msg.delete()
await ctx.reply('{} message(s) transferred to {}.'.format(len(saved), destination.name))
def get_message(self, guild):
with self.cursor_context() as cursor:
res = cursor.execute(*db_util.select("welcomes", items="message").where(id_server=guild.id).build)
if not res:
cursor.execute(*db_util.insert("welcomes").items(id_server=guild.id, message="").build)
return None
row = cursor.fetchone()
text = row[0]
return text
async def name(self, ctx):
with self.cursor_context(commit=True) as cursor:
cursor.execute(*db_util.select("event_logs").items("id_user")
.where(id_server=325197025719091201).distinct.build)
rows = cursor.fetchall()
if rows:
with self.cursor_context(commit=True) as cursor:
for row in rows:
user = await self.bot.get_user_info(row[0])
if user is None:
continue
cursor.execute(*db_util.insert("users", id_user=user.id, user_name=str(user)))
await ctx.author.send('Done')
async def on_member_unban(self, guild, user):
with self.cursor_context(commit=True) as cursor:
db_util.insert("event_logs").items(id_server=guild.id,
id_user=user.id,
date_utc=DBFunction("NOW()"),
event_type="left").run(cursor)