forked from axiom-team/astroport
126 lines
3.9 KiB
Python
126 lines
3.9 KiB
Python
import sys, re, json, requests, base64
|
|
from time import time
|
|
from lib.cesiumCommon import CesiumCommon, PUBKEY_REGEX
|
|
|
|
|
|
class Profiles(CesiumCommon):
|
|
# Configure JSON document SET to send
|
|
def configDocSet(self, name, description, city, address, pos, socials, avatar):
|
|
timeSent = int(time())
|
|
|
|
data = {}
|
|
if name: data['title'] = name
|
|
if description: data['description'] = description
|
|
if address: data['address'] = address
|
|
if city: data['city'] = city
|
|
if pos:
|
|
geoPoint = {}
|
|
geoPoint['lat'] = pos[0]
|
|
geoPoint['lon'] = pos[1]
|
|
data['geoPoint'] = geoPoint
|
|
if socials:
|
|
data['socials'] = []
|
|
data['socials'].append({})
|
|
data['socials'][0]['type'] = "web"
|
|
data['socials'][0]['url'] = socials
|
|
if avatar:
|
|
avatar = open(avatar, 'rb').read()
|
|
avatar = base64.b64encode(avatar).decode()
|
|
data['avatar'] = {}
|
|
data['avatar']['_content'] = avatar
|
|
data['avatar']['_content_type'] = "image/png"
|
|
data['time'] = timeSent
|
|
data['issuer'] = self.pubkey
|
|
data['version'] = 2
|
|
data['tags'] = []
|
|
|
|
document = json.dumps(data)
|
|
|
|
return self.signDoc(document)
|
|
|
|
# Configure JSON document GET to send
|
|
def configDocGet(self, profile, scope='title', getAvatar=None):
|
|
|
|
if getAvatar:
|
|
avatar = "avatar"
|
|
else:
|
|
avatar = "avatar._content_type"
|
|
|
|
data = {
|
|
"query": {
|
|
"bool": {
|
|
"should":[
|
|
{
|
|
"match":{
|
|
scope:{
|
|
"query": profile,"boost":2
|
|
}
|
|
}
|
|
},{
|
|
"prefix": {scope: profile}
|
|
}
|
|
]
|
|
}
|
|
},"highlight": {
|
|
"fields": {
|
|
"title":{},
|
|
"tags":{}
|
|
}
|
|
},"from":0,
|
|
"size":100,
|
|
"_source":["title", avatar,"description","city","address","socials.url","creationTime","membersCount","type"],
|
|
"indices_boost":{"user":100,"page":1,"group":0.01
|
|
}
|
|
}
|
|
|
|
document = json.dumps(data)
|
|
|
|
return document
|
|
|
|
# Configure JSON document SET to send
|
|
def configDocErase(self):
|
|
timeSent = int(time())
|
|
|
|
data = {}
|
|
data['time'] = timeSent
|
|
data['id'] = self.pubkey
|
|
data['issuer'] = self.pubkey
|
|
data['version'] = 2
|
|
data['index'] = "user"
|
|
data['type'] = "profile"
|
|
|
|
document = json.dumps(data)
|
|
|
|
return self.signDoc(document)
|
|
|
|
def sendDocument(self, document, type):
|
|
|
|
headers = {
|
|
'Content-type': 'application/json',
|
|
}
|
|
|
|
# Send JSON document and get JSON result
|
|
if type == 'set':
|
|
reqQuery = '{0}/user/profile?pubkey={1}/_update?pubkey={1}'.format(self.pod, self.pubkey)
|
|
elif type == 'get':
|
|
reqQuery = '{0}/user,page,group/profile,record/_search'.format(self.pod)
|
|
elif type == 'erase':
|
|
reqQuery = '{0}/history/delete'.format(self.pod)
|
|
|
|
result = requests.post(reqQuery, headers=headers, data=document)
|
|
if result.status_code == 200:
|
|
# print(result.text)
|
|
return result.text
|
|
else:
|
|
sys.stderr.write("Echec de l'envoi du document...\n" + result.text + '\n')
|
|
|
|
def parseJSON(self, doc):
|
|
doc = json.loads(doc)['hits']['hits']
|
|
if doc:
|
|
pubkey = { "pubkey": doc[0]['_id'] }
|
|
rest = doc[0]['_source']
|
|
final = {**pubkey, **rest}
|
|
return json.dumps(final, indent=2)
|
|
else:
|
|
return 'Profile vide'
|