aht_bridge/utils_AHT.py at main · robervez/aht_bridge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import json
import requests
import configparser


class AHT():

    def __init__(self, ahip):
        self.AHbaseURL= ahip

    def unregisterService(self):

        srvdef = aht.findService('addfacetofacemaskdb')

        if srvdef is not None:
            id=srvdef.get('id',None)
            if id is not None:

                url = self.AHbaseURL + '/serviceregistry/mgmt/' + str(id)
                myrequest = requests.delete(url)
                myresponse = myrequest.status_code
                if myresponse==200:
                    return True
                else:
                    print(myresponse)
                    return False


    def registerService(self, servername):

        url = self.AHbaseURL + '/serviceregistry/mgmt'

        headers = {'Content-type': 'application/json', 'Accept': 'application/json'}
        dataPOST={"endOfValidity": "2022-03-31 16:29:53",
                  "interfaces": ["HTTP-SECURE-JSON"],
                  "metadata": {
                    "Company": "UniMoRe",
                    "Author": "Roberto Vezzani",
                    "Description": "Add a face to the db for the annotation of mask/notMask"
                  },
                  "providerSystem": {
                    "address": servername,
                    "authenticationInfo": "fweWf24exc.#234",
                    "port": 443,
                    "systemName": "AHT-MoRe-Masks"
                  },
                  "secure": "NOT_SECURE",
                  "serviceDefinition": "AddFaceToFaceMaskDB",
                  "serviceUri": "uploadimagewithkey",
                  "version": 0
                }


        myrequest = requests.post(url, json=dataPOST,  headers=headers)
        myresponse = myrequest.text

        print(myresponse)

    def getListOfServices(self):
        listService=[]
        url = self.AHbaseURL + '/serviceregistry/mgmt?direction=ASC&sort_field=id'
        headers = dict()
        headers['Accept']='*/*'

        myrequest = requests.get (url,  headers=headers)
        dataget = myrequest.text
        fileOut= open('services.txt','w')
        fileOut.write(dataget )
        fileOut.close()
        print(dataget)

    def findService(self, name):
        listService = []
        url = self.AHbaseURL + '/serviceregistry/mgmt?direction=ASC&sort_field=id'
        headers = dict()
        headers['Accept'] = '*/*'

        myrequest = requests.get(url, headers=headers)
        dataget = myrequest.text
        jsonservices = json.loads(dataget)
        for serv in jsonservices['data']:
            srvdef = serv.get('serviceDefinition',dict())
            if srvdef.get('serviceDefinition','') == name:
                # servizio trovato
                return serv
        return None

def geturlFromSrcdef(srvdef):
    myurl='nd'
    if srvdef is None:
        return ''
    provider = srvdef.get('provider',None)
    if provider['port']==443:
        myurl = 'https://'
    else:
        myurl = 'http://'
    myurl += provider['address']
    myurl +='/'
    myurl += srvdef.get('serviceUri')
    #if (srvdef.get['servi'])

    return myurl

if __name__=='__main__':
    Config = configparser.ConfigParser()
    conf = Config.read("gateway.ini")


    aht = AHT(Config.get('General','BASEURL'))
    #print (aht.getListOfServices())
    srvdef = aht.findService('addfacetofacemaskdb')
    print('url service before reg: ' + geturlFromSrcdef(srvdef))

    aht.unregisterService()

    aht.registerService(Config.get('FaceMask','SERVER'))

    srvdef = aht.findService('addfacetofacemaskdb')
    print('url service after reg: ' + geturlFromSrcdef(srvdef))