96 lines
4.3 KiB
Python
96 lines
4.3 KiB
Python
import datetime
|
|
import json
|
|
import os
|
|
import pytz
|
|
|
|
|
|
async def import_old_database(bot, server):
|
|
"""try to import the old database"""
|
|
try:
|
|
clean_server = str(server).replace("/", "")
|
|
while clean_server.startswith("."):
|
|
clean_server = clean_server[1:]
|
|
fn = 'backup/' + clean_server + '.json'
|
|
with open(fn, 'r') as infile:
|
|
polls = json.load(infile)
|
|
for p in polls:
|
|
#print(polls[p]['short'])
|
|
wr = []
|
|
wn = []
|
|
for r, n in polls[p]['weights'].items():
|
|
wr.append(r)
|
|
try:
|
|
if n.is_integer():
|
|
n = int(n)
|
|
except:
|
|
pass
|
|
wn.append(n)
|
|
created = datetime.datetime.strptime(polls[p]['datestarted'], '%d-%m-%Y %H:%M').replace(tzinfo=pytz.utc)
|
|
if polls[p]['duration'] == 0:
|
|
duration = 0
|
|
else:
|
|
duration = created + datetime.timedelta(hours=float(polls[p]['duration']))
|
|
votes = {}
|
|
for u,o in polls[p]['votes'].items():
|
|
# get weight
|
|
user = server.get_member(u)
|
|
weight = 1
|
|
if wr.__len__() > 0:
|
|
valid_weights = [wn[wr.index(r)] for r in
|
|
list(set([n.name for n in user.roles]).intersection(set(wr)))]
|
|
if valid_weights.__len__() > 0:
|
|
#print(wr, wn)
|
|
weight = max(valid_weights)
|
|
choices = []
|
|
if o in polls[p]['options']:
|
|
choices = [polls[p]['options'].index(o)]
|
|
votes[u] = {'weight': weight, 'choices': choices}
|
|
|
|
new_format = {
|
|
'server_id': str(server.id),
|
|
'channel_id': str(polls[p]['channel']),
|
|
'author': str(polls[p]['author']),
|
|
'name': polls[p]['name'],
|
|
'short': polls[p]['short'],
|
|
'anonymous': polls[p]['anonymous'],
|
|
'reaction': True,
|
|
'multiple_choice': False,
|
|
'options_reaction': polls[p]['options'],
|
|
'reaction_default': False,
|
|
'roles': polls[p]['roles'],
|
|
'weights_roles': wr,
|
|
'weights_numbers': wn,
|
|
'duration': duration,
|
|
'duration_tz': 'UTC',
|
|
'time_created': created,
|
|
'open': polls[p]['open'],
|
|
'active': True,
|
|
'activation': 0,
|
|
'activation_tz': 'UTC',
|
|
'votes': votes
|
|
}
|
|
await bot.db.polls.update_one({'server_id': str(server.id), 'short': polls[p]['short']},
|
|
{'$set': new_format}, upsert=True)
|
|
os.remove(fn)
|
|
except FileNotFoundError:
|
|
pass
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
# potential update message
|
|
# text = "**Dear Server Admin!**\n\n" \
|
|
# "After more than a year in the field, today Pollmaster received it's first big update and I am excited to present you the new Version!\n\n" \
|
|
# "**TL;DR: A massive overhaul of every function. The new (now customizable) prefix is `pm!` and you can find the rest of the commands with `pm!help`**\n\n" \
|
|
# "**Here are some more highlights:**\n" \
|
|
# "🔹 Voting is no longer done per text, but by using reactions\n" \
|
|
# "🔹 Creating new polls is now an interactive process instead of command lines\n" \
|
|
# "🔹 There is now a settings for multiple choice polls\n" \
|
|
# "🔹 You can use all the commands in a private message with Pollmaster to reduce spam in your channels\n\n" \
|
|
# "For the full changelog, please visit: https://github.com/matnad/pollmaster/blob/master/changelog.md\n" \
|
|
# "All your old polls should be converted to the new format and accessible with `pm!show` or `pm!show closed`.\n" \
|
|
# "If you have any problems or questions, please join the support discord: https://discord.gg/Vgk8Nve\n\n" \
|
|
# "**No action from you is required. But you might want to update permissions for your users. " \
|
|
# "See pm!help -> configuration**\n\n" \
|
|
# "I hope you enjoy the new Pollmaster!\n" \
|
|
# "Regards, Newti#0654" |