Vehicle-Anti-Theft-Face-Recognition-System/DBHelper.py

import pyrebase

firebaseConfig = {
    'apiKey': "AIzaSyAdL0W5HscjEDFPK4BDi6Cnc7FLa30GPYY",
    'authDomain': "vehicleantitheftrecognition.firebaseapp.com",
    'databaseURL': "https://vehicleantitheftrecognition.firebaseio.com",
    'projectId': "vehicleantitheftrecognition",
    'storageBucket': "vehicleantitheftrecognition.appspot.com",
    'messagingSenderId': "163692530359",
    'appId': "1:163692530359:web:b6dc7ccfc56a79afb11b32",
    'measurementId': "G-EPWP2LK89Q",
    'serviceAccount': 'vehicleantitheftrecognition-firebase-adminsdk-krrgw-05da515de5.json'
}

firebase = pyrebase.initialize_app(firebaseConfig)
db = firebase.database()
auth = firebase.auth()
storage = firebase.storage()


# Create account function which creates a new authentication info.
def create_account(username, password, confirmpassword):
    email = username + "@hotmail.com"
    if password == confirmpassword:
        auth.create_user_with_email_and_password(email, password)
        print("Account sucessfully created.")
    else:
        print("Confirmed password doesn't match to other password.")


# Login function which verifies the given authentication info.
def login(username, password):
    email = username + "@hotmail.com"
    try:
        auth.sign_in_with_email_and_password(email, password)
        print("Successfully Logged in.")
    except:
        print("Invalid username or password.")


# Uploads the data of specified user uploaded into firebase.
def upload_data(userID, firstname, lastname, email, phone, address):
    data = {"First Name": firstname, "Last Name": lastname, "E-Mail": email, "Phone": phone, "Address": address}
    db.child("Users").child(userID).set(data)


# Removes the data of specified user uploaded into firebase.
def remove_data(userID):
    db.child("Users").child(userID).remove()


# Returns the first name or else an empty string.
def get_firstname(userID):
    firstname = ""
    users = db.child("Users").get()
    for user in users.each():
        if user.key() == userID:
            firstname = user.val()["First Name"]
    return firstname


# Returns the last name or else an empty string.
def get_lastname(userID):
    lastname = ""
    users = db.child("Users").get()
    for user in users.each():
        if user.key() == userID:
            lastname = user.val()["Last Name"]
    return lastname


# Returns the e-mail or else an empty string.
def get_email(userID):
    email = ""
    users = db.child("Users").get()
    for user in users.each():
        if user.key() == userID:
            email = user.val()["E-Mail"]
    return email


# Returns the phone or else an empty string.
def get_phone(userID):
    phone = ""
    users = db.child("Users").get()
    for user in users.each():
        if user.key() == userID:
            phone = user.val()["Phone"]
    return phone


# Returns the address or else an empty string.
def get_address(userID):
    address = ""
    users = db.child("Users").get()
    for user in users.each():
        if user.key() == userID:
            address = user.val()["Address"]
    return address


# Uploads the photo of user, input should be something like "example.png"
def upload_user_photo(userphoto):
    userphoto_str = str(userphoto)
    storage.child("Photos_of_Users/" + str(userphoto)).put("Photos_of_Users/" + str(userphoto))


# Uploads the photo of thief, input should be something like "example.png"
def upload_thief_photo(userphoto):
    userphoto_str = str(userphoto)
    storage.child("Photos_of_Thieves/" + str(userphoto)).put("Photos_of_Thieves/" + str(userphoto))


# Downloads all the user photos.
def download_all_user_photos(self):
    storage.child("Photos_of_Users").download("Storage_from_Database")


# Downloads all the thief photos.
def download_all_thief_photos(self):
    storage.child("Photos_of_Thieves").download("Storage_from_Thieves")


# Deletes photo of the specified user.
def delete_user_photo(userphoto):
    storage.delete('Photos_of_Users/' + userphoto)