mirror of https://gitlab.com/bashrc2/epicyon
100 lines
3.7 KiB
Python
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)
|