epicyon/question.py

__filename__ = "question.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.2.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"
__module_group__ = "ActivityPub"

import os
from utils import locatePost
from utils import loadJson
from utils import saveJson
from utils import hasObjectDict


def questionUpdateVotes(baseDir: str, nickname: str, domain: str,
                        replyJson: {}) -> {}:
    """ For a given reply update the votes on a question
    Returns the question json object if the vote totals were changed
    """
    if not hasObjectDict(replyJson):
        return None
    if not replyJson['object'].get('inReplyTo'):
        return None
    if not replyJson['object']['inReplyTo']:
        return None
    if not isinstance(replyJson['object']['inReplyTo'], str):
        return None
    if not replyJson['object'].get('name'):
        return None
    inReplyTo = replyJson['object']['inReplyTo']
    questionPostFilename = locatePost(baseDir, nickname, domain, inReplyTo)
    if not questionPostFilename:
        return None
    questionJson = loadJson(questionPostFilename)
    if not questionJson:
        return None
    if not hasObjectDict(questionJson):
        return None
    if not questionJson['object'].get('type'):
        return None
    if questionJson['type'] != 'Question':
        return None
    if not questionJson['object'].get('oneOf'):
        return None
    if not isinstance(questionJson['object']['oneOf'], list):
        return None
    if not questionJson['object'].get('content'):
        return None
    replyVote = replyJson['object']['name']
    # does the reply name field match any possible question option?
    foundAnswer = None
    for possibleAnswer in questionJson['object']['oneOf']:
        if not possibleAnswer.get('name'):
            continue
        if possibleAnswer['name'] == replyVote:
            foundAnswer = possibleAnswer
            break
    if not foundAnswer:
        return None
    # update the voters file
    votersFileSeparator = ';;;'
    votersFilename = questionPostFilename.replace('.json', '.voters')
    if not os.path.isfile(votersFilename):
        # create a new voters file
        with open(votersFilename, 'w+') as votersFile:
            votersFile.write(replyJson['actor'] +
                             votersFileSeparator +
                             foundAnswer + '\n')
    else:
        if replyJson['actor'] not in open(votersFilename).read():
            # append to the voters file
            with open(votersFilename, 'a+') as votersFile:
                votersFile.write(replyJson['actor'] +
                                 votersFileSeparator +
                                 foundAnswer + '\n')
        else:
            # change an entry in the voters file
            with open(votersFilename, "r") as votersFile:
                lines = votersFile.readlines()
                newlines = []
                saveVotersFile = False
                for voteLine in lines:
                    if voteLine.startswith(replyJson['actor'] +
                                           votersFileSeparator):
                        newVoteLine = replyJson['actor'] + \
                            votersFileSeparator + foundAnswer + '\n'
                        if voteLine == newVoteLine:
                            break
                        saveVotersFile = True
                        newlines.append(newVoteLine)
                    else:
                        newlines.append(voteLine)
                if saveVotersFile:
                    with open(votersFilename, 'w+') as votersFile:
                        for voteLine in newlines:
                            votersFile.write(voteLine)
                else:
                    return None
    # update the vote counts
    questionTotalsChanged = False
    for possibleAnswer in questionJson['object']['oneOf']:
        if not possibleAnswer.get('name'):
            continue
        totalItems = 0
        with open(votersFilename, "r") as votersFile:
            lines = votersFile.readlines()
            for voteLine in lines:
                if voteLine.endswith(votersFileSeparator +
                                     possibleAnswer['name'] + '\n'):
                    totalItems += 1
        if possibleAnswer['replies']['totalItems'] != totalItems:
            possibleAnswer['replies']['totalItems'] = totalItems
            questionTotalsChanged = True
    if not questionTotalsChanged:
        return None
    # save the question with altered totals
    saveJson(questionJson, questionPostFilename)
    return questionJson


def isQuestion(postObjectJson: {}) -> bool:
    """ is the given post a question?
    """
    if postObjectJson['type'] != 'Create' and \
       postObjectJson['type'] != 'Update':
        return False
    if not hasObjectDict(postObjectJson):
        return False
    if not postObjectJson['object'].get('type'):
        return False
    if postObjectJson['object']['type'] != 'Question':
        return False
    if not postObjectJson['object'].get('oneOf'):
        return False
    if not isinstance(postObjectJson['object']['oneOf'], list):
        return False
    return True