Voting.py

import datetime
import time
from EmojiListener import EmojiListener


def getTimeStamp():
    return "[VOTING] [" + time.strftime('%Y-%m-%d %H:%M:%S') + "] "


async def create_archived_votes(client):
    guild_file = "data/" + str(client.guild.id) + "_emojivotes.max"
    try:
        file = open(guild_file, "r")
        raw = file.read()
        raw_voters = raw.split('\n')
        out = []
        for voter in raw_voters:
            if voter != "":
                voter = voter.split(",")
                created_voter = await create_voter(voter, client)
                if created_voter is not None and not created_voter.get_passed():
                    out.append(created_voter)
        file.close()
        return out
    except FileNotFoundError:
        file = open(guild_file, "x")
        file.close()
        return []


async def create_voter(raw_voter, client):
    try:
        message_channel = await client.fetch_channel(int(raw_voter[1]))
        message = await message_channel.fetch_message(int(raw_voter[0]))

        announcements_channel = await client.fetch_channel(raw_voter[3])
        voter_message = await announcements_channel.fetch_message(raw_voter[2])

        emoji = EmojiListener(message, announcements_channel, message.guild, voter_message)
        await emoji.handle_image_silent()

        # TODO make this "15" automatically change the with the voting threshold
        return VotingListener(emoji, 15, client, False)

    except Exception as e:
        print(getTimeStamp(), "Failed to find message")
        return None


class VotingListener:

    def __init__(self, emoji, thresh, client, new_voter):
        self.client = client
        self.threshold = thresh
        self.emoji = emoji
        self.active = True
        self.author_id = emoji.message.author.id
        if new_voter:
            self.write_voter_to_file()
            print(getTimeStamp() + "Created Emoji Voter:", emoji.name)
        else:
            print(getTimeStamp() + "Recovered Voter:", emoji.name)
        # self.update_user_submission_record()

    # FORMAT OF WRITTEN FILES:
    # original submission message ID, channel ID of <-, voting message id, channel ID of <-
    def write_voter_to_file(self):
        guild_file = "data/" + str(self.client.guild.id) + "_emojivotes.max"
        file = open(guild_file, "a")
        out = str(self.emoji.message.id) + "," + str(self.emoji.message.channel.id) + "," + \
              str(self.emoji.voter_message.id) + "," + str(self.emoji.voter_message.channel.id)
        file.write(out+"\n")
        file.close()

    def get_passed(self):
        for reaction in self.emoji.voter_message.reactions:
            if isinstance(reaction.emoji, str):
                print(reaction.emoji)
                if reaction.emoji.__contains__("posrep"):
                    if reaction.count >= self.threshold:
                        return True
            else:
                if reaction.emoji.name == "posrep":
                    if reaction.count >= self.threshold:
                        return True
        return False

    def get_message_id(self):
        if self.active:
            s = self.emoji.voter_message
            return s.id

    async def add_vote(self, reaction):
        channel = self.client.get_channel(reaction.channel_id)
        message = await channel.fetch_message(reaction.message_id)
        for reaction in message.reactions:
            if reaction.emoji.name == "posrep":
                print(getTimeStamp() + self.emoji.name, "Votes:", reaction.count)
                if reaction.count >= self.threshold:
                    return True, self.emoji.image.bytes, self.emoji.name, self.emoji.toBeReplaced
                else:
                    break
        return False, None, None, None

    def is_active(self):
        return self.emoji.in_use

    def destroy(self):
        self.active = False

    def update_user_submission_record(self):
        guild_file = "data/" + str(self.client.guild.id) + "_emojivotes.max"
        file = str(open(guild_file, "r").read()).split("\n")
        out = ""
        for entry in file:
            id = entry.split(";")[0]
            if id != self.emoji.message.author.id:
                out += entry + "\n"
        out += str(self.emoji.message.author.id) + ";" + str(time.time()) + "\n"
        file = open(guild_file, "w")
        file.write(out)
        file.close()