epicyon/question.py

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

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

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 replyJson.get('object'):
        return None
    if not isinstance(replyJson['object'], dict):
        return None
    if not replyJson['object'].get('inReplyTo'):
        return None
    if not replyJson['object']['inReplyTo']:
        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 questionJson.get('object'):
        return None
    if not isinstance(questionJson['object'], dict):
        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
        votersFile=open(votersFilename, "w")
        if votersFile:
            votersFile.write(replyJson['actor']+votersFileSeparator+foundAnswer+'\n')
            votersFile.close()
    else:
        if replyJson['actor'] not in open(votersFilename).read():
            # append to the voters file
            votersFile=open(votersFilename, "a+")
            if votersFile:
                votersFile.write(replyJson['actor']+votersFileSeparator+foundAnswer+'\n')
                votersFile.close()
        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