Пример #1
0
def listG():
    query = 'SELECT groups.*, subgroups.subgroupID, subgroups.subgroupName FROM groups LEFT JOIN subgroups ON groups.groupID=subgroups.groupID'
    # try:
    cur.execute(query)
    temp = cur.fetchall()
    # except Error as E:
    #     return jsonify({'error' : E})
    return jsonify(temp)
Пример #2
0
def enter():
    req = request.json
    query = 'INSERT INTO userinsubgroup VALUES (%s,%s)'
    try:
        cur.execute(query, (req['id'], req['subgroup']))
        con.commit()
    except Error as E:
        return E
    return "Success"
Пример #3
0
def addG():
    req = request.json
    name = req['groupname']
    query = f"INSERT INTO groups(groupName) VALUES('{name}')"
    try:
        cur.execute(query)
        con.commit()
    except Error as E:
        return jsonify({'error': str(E)})
    return jsonify({"user": get_jwt_identity(), "result": "Succeed"})
Пример #4
0
def addSG():
    req = request.json
    name = req['subgroupname']
    groupId = req['groupID']
    query = 'INSERT INTO subgroups(groupID,subgroupName) VALUES (%s,%s)'
    try:
        cur.execute(query, (groupId, name))
        con.commit()
    except Error as E:
        return jsonify({'error': E})
    return "suceed"
Пример #5
0
def deleteface():
    req = request.args
    try:
        query2 = f"DELETE FROM images WHERE id={req['id']}"
        query = f"DELETE FROM encoding WHERE faceID={req['id']}"
        cur.execute(query2)
        cur.execute(query)
    except Error as E:
        print(E)
        return 'Err'
    con.commit()
    return "Succeed "
Пример #6
0
def selectFaces():
    req = request.args
    try:
        query = f"SELECT faceID, faceOwner FROM encoding WHERE subgroupID={req['subgrouid']}"
        cur.execute(query)
    except Error as E:
        print(E)
        return 'Err'
    data = cur.fetchall()
    mylist = []
    for i in data:
        mylist.append(i)
    return jsonify({'Result': mylist})
Пример #7
0
def sign():
    req = request.json
    uname = req['userid']
    role = req['role']
    if role.lower() not in ['superuser', 'user']:
        return jsonify({'error': "invalid role"})
    query = 'INSERT INTO user (username,role) VALUES (%s,%s)'
    try:
        cur.execute(query, (uname, role))
        con.commit()
    except Error as E:
        # print(E)
        return 500, jsonify({"error": str(E)})
    return "succeed"
Пример #8
0
def recognize():
    req = request.form
    imgfile = request.files['image']
    try:
        query = f"SELECT faceOwner,encodingblob FROM encoding WHERE subgroupID={req['subgroupid']}"
        cur.execute(query)
    except Error as E:
        # print(E)
        return str(E)
    x = cur.fetchall()
    knownEncodings = []
    knownNames = []
    for i in x:
        res = compare(imgfile, [fromBlob(i[1])], [i[0]])
        if res != 'unknown name':
            return res
    return jsonify({'Result': res})
Пример #9
0
def byteface():
    idF = request.args['id']
    if client.bucket_exists('facerecimages'):
        # try:
        query = f"SELECT path FROM images WHERE id={idF}"
        cur.execute(query)
        filetype = cur.fetchall()[0][0]
        if not filetype:
            return "id not found"
        resp = client.get_object('facerecimages', str(idF))
        byt = resp.data
        # print(byt)

        # KALO MAU NGESAVE JADI FILE LOCAL
        # stream = BytesIO(byt)
        # img = Image.open(stream)
        # img.save('test.jpg')
    else:
        return jsonify({"error": "bucket not found"})
    return jsonify({"filetype": filetype, "b64data": str(byt)})
Пример #10
0
def saveFile(idF, imgfile, upload_folder, client):
    # if allowed(file.name):
    # filetype = imgfile.name.split('.')[-1]
    # images tadinya ga kepake, skrg dibuat nyimpen filetype
    query = "INSERT INTO images(id, faceOwner, path) VALUES(%s,%s,%s)"
    cur.execute(query,(idF,"",imgfile.filename.split('.')[-1]))
    con.commit()
    bytestr = imgfile.read()
    # stream = BytesIO(bytestr)
    streamdata = b64encode(str(bytestr).encode('utf-8'))
    stream = BytesIO(streamdata)
    # stream = StringIO(streamdata)
    if client.bucket_exists("facerecimages"):
        pass
    else:
        client.make_bucket("facerecimages")
    client.put_object("facerecimages", str(idF), stream, len(bytestr))
    # imgfile.save(path)
    # else:
        # raise jsonify({"Error" : })
        
Пример #11
0
def enroll():
    # role = 'Superuser'
    req = request.form
    print(req)
    user = req['userid']
    imgfile = request.files['image']
    # print(imgfile.filename)
    # filetype = req['filename'].split('.')[-1]
    try:
        query = f"SELECT role FROM user WHERE username='******'"
        cur.execute(query)
    except Error as E:
        return str(E)
    role = cur.fetchall()
    if role[0][0] != 'superuser':
        return jsonify({"error": 401, "reason": "Invalid role"})
    query = 'INSERT INTO encoding (subgroupID, faceOwner, encodingblob) VALUES (%s,%s,%s)'
    query2 = "UPDATE encoding SET encodingblob=%s WHERE faceID=%s"
    try:
        cur.execute(query, (req['subgroupid'], req['name'], ""))
        lastid = cur.lastrowid
        saveFile(lastid, imgfile, UPLOAD_FOLDER, client)
        imgfile.seek(0)
        x = makeBlob(imgfile)
        cur.execute(query2, (x, lastid))
        con.commit()
        return jsonify({"last face id": lastid})
    except Error as E:
        return jsonify({'error': str(E)})