epicyon/question.py

100 lines
3.7 KiB
Python

__filename__ = "question.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.0.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
"""
if not replyJson.get('object'):
return
if not isinstance(replyJson['object'], dict):
return
if not replyJson['object'].get('inReplyTo'):
return
if not replyJson['object']['inReplyTo']:
return
if not replyJson['object'].get('content'):
return
inReplyTo=replyJson['object']['inReplyTo']
questionPostFilename=locatePost(baseDir,nickname,domain,inReplyTo)
if not questionPostFilename:
return
questionJson=loadJson(questionPostFilename)
if not questionJson:
return
if not questionJson.get('object'):
return
if not isinstance(questionJson['object'], dict):
return
if not questionJson['object'].get('type'):
return
if questionJson['type']!='Question':
return
if not questionJson['object'].get('oneOf'):
return
if not isinstance(questionJson['object']['oneOf'], list):
return
content=replyJson['object']['content']
# does the reply content match any possible question option?
foundAnswer=None
for possibleAnswer in questionJson['object']['oneOf']:
if not possibleAnswer.get('name'):
continue
if possibleAnswer['name'] in content:
foundAnswer=possibleAnswer
break
if not foundAnswer:
return
# 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()
with open(votersFilename, "w") as votersFile:
for voteLine in lines:
if voteLine.startswith(replyJson['actor']+votersFileSeparator):
votersFile.write(replyJson['actor']+votersFileSeparator+foundAnswer+'\n')
break
votersFile.write(voteLine)
# 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 questionTotalsChanged:
# save the question with altered totals
saveJson(questionJson,questionPostFilename)