epicyon/skills.py

167 lines
5.3 KiB
Python
Raw Normal View History

2019-07-19 10:01:24 +00:00
__filename__ = "skills.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
2019-08-29 13:35:29 +00:00
__version__ = "1.0.0"
2019-07-19 10:01:24 +00:00
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"
import json
import commentjson
import os
2019-10-12 09:37:21 +00:00
import time
2019-07-19 10:01:24 +00:00
from webfinger import webfingerHandle
from auth import createBasicAuthHeader
from posts import getPersonBox
from session import postJson
from utils import getNicknameFromActor
from utils import getDomainFromActor
2019-10-22 11:55:06 +00:00
from utils import loadJson
from utils import saveJson
2019-07-19 10:01:24 +00:00
def setSkillLevel(baseDir: str,nickname: str,domain: str, \
skill: str,skillLevelPercent: int) -> bool:
"""Set a skill level for a person
Setting skill level to zero removes it
"""
if skillLevelPercent<0 or skillLevelPercent>100:
return False
actorFilename=baseDir+'/accounts/'+nickname+'@'+domain+'.json'
if not os.path.isfile(actorFilename):
return False
2019-09-30 22:39:02 +00:00
2019-10-22 11:55:06 +00:00
actorJson=loadJson(actorFilename)
2019-09-30 22:39:02 +00:00
if actorJson:
2019-07-19 10:01:24 +00:00
if not actorJson.get('skills'):
actorJson['skills']={}
if skillLevelPercent>0:
actorJson['skills'][skill]=skillLevelPercent
else:
del actorJson['skills'][skill]
2019-10-22 11:55:06 +00:00
saveJson(actorJson,actorFilename)
2019-07-19 10:01:24 +00:00
return True
2019-08-09 08:46:38 +00:00
def setSkills(baseDir: str,nickname: str,domain: str,skills: {}) -> None:
actorFilename=baseDir+'/accounts/'+nickname+'@'+domain+'.json'
if not os.path.isfile(actorFilename):
return False
2019-09-30 22:39:02 +00:00
2019-10-22 11:55:06 +00:00
actorJson=loadJson(actorFilename)
2019-09-30 22:39:02 +00:00
if actorJson:
2019-08-09 08:46:38 +00:00
actorJson['skills']=skills
2019-10-22 11:55:06 +00:00
saveJson(actorJson,actorFilename)
2019-08-09 08:46:38 +00:00
2019-07-19 10:01:24 +00:00
def getSkills(baseDir: str,nickname: str,domain: str) -> []:
"""Returns the skills for a given person
"""
actorFilename=baseDir+'/accounts/'+nickname+'@'+domain+'.json'
if not os.path.isfile(actorFilename):
return False
2019-09-30 22:39:02 +00:00
2019-10-22 11:55:06 +00:00
actorJson=loadJson(actorFilename)
2019-09-30 22:39:02 +00:00
if actorJson:
2019-07-19 10:01:24 +00:00
if not actorJson.get('skills'):
return None
return actorJson['skills']
return None
def outboxSkills(baseDir: str,nickname: str,messageJson: {},debug: bool) -> bool:
"""Handles receiving a skills update
"""
if not messageJson.get('type'):
return False
if not messageJson['type']=='Skill':
return False
if not messageJson.get('actor'):
return False
if not messageJson.get('object'):
return False
if not isinstance(messageJson['object'], str):
return False
actorNickname=getNicknameFromActor(messageJson['actor'])
if actorNickname!=nickname:
return False
domain,port=getDomainFromActor(messageJson['actor'])
2019-07-19 11:38:37 +00:00
skill=messageJson['object'].replace('"','').split(';')[0].strip()
skillLevelPercent=int(messageJson['object'].replace('"','').split(';')[1].strip())
2019-07-19 10:01:24 +00:00
return setSkillLevel(baseDir,nickname,domain, \
skill,skillLevelPercent)
2019-08-20 09:16:03 +00:00
def sendSkillViaServer(baseDir: str,session,nickname: str,password: str,
2019-07-19 10:01:24 +00:00
domain: str,port: int, \
httpPrefix: str, \
skill: str,skillLevelPercent: int, \
cachedWebfingers: {},personCache: {}, \
2019-08-14 20:12:27 +00:00
debug: bool,projectVersion: str) -> {}:
2019-07-19 10:01:24 +00:00
"""Sets a skill for a person via c2s
"""
if not session:
print('WARN: No session for sendSkillViaServer')
return 6
domainFull=domain
if port:
if port!=80 and port!=443:
if ':' not in domain:
domainFull=domain+':'+str(port)
2019-07-19 10:01:24 +00:00
toUrl = httpPrefix+'://'+domainFull+'/users/'+nickname
2019-07-19 11:38:37 +00:00
ccUrl = httpPrefix+'://'+domainFull+'/users/'+nickname+'/followers'
2019-07-19 10:01:24 +00:00
if skillLevelPercent:
skillStr=skill+';'+str(skillLevelPercent)
else:
skillStr=skill+';0'
newSkillJson = {
'type': 'Skill',
'actor': httpPrefix+'://'+domainFull+'/users/'+nickname,
2019-07-19 11:38:37 +00:00
'object': '"'+skillStr+'"',
2019-07-19 10:01:24 +00:00
'to': [toUrl],
'cc': [ccUrl]
}
handle=httpPrefix+'://'+domainFull+'/@'+nickname
# lookup the inbox for the To handle
2019-08-14 20:12:27 +00:00
wfRequest = webfingerHandle(session,handle,httpPrefix,cachedWebfingers, \
domain,projectVersion)
2019-07-19 10:01:24 +00:00
if not wfRequest:
if debug:
print('DEBUG: announce webfinger failed for '+handle)
return 1
postToBox='outbox'
# get the actor inbox for the To handle
inboxUrl,pubKeyId,pubKey,fromPersonId,sharedInbox,capabilityAcquisition,avatarUrl,displayName = \
2019-08-20 09:16:03 +00:00
getPersonBox(baseDir,session,wfRequest,personCache, \
2019-10-17 15:55:05 +00:00
projectVersion,httpPrefix,nickname,domain,postToBox)
2019-07-19 10:01:24 +00:00
if not inboxUrl:
if debug:
print('DEBUG: No '+postToBox+' was found for '+handle)
return 3
if not fromPersonId:
if debug:
print('DEBUG: No actor was found for '+handle)
return 4
authHeader=createBasicAuthHeader(Nickname,password)
headers = {'host': domain, \
'Content-type': 'application/json', \
'Authorization': authHeader}
postResult = \
postJson(session,newSkillJson,[],inboxUrl,headers,"inbox:write")
#if not postResult:
# if debug:
# print('DEBUG: POST announce failed for c2s to '+inboxUrl)
# return 5
if debug:
print('DEBUG: c2s POST skill success')
return newSkillJson