예제 #1
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def show_users():
    f = open(folder + 'database.db', 'rb')
    user_data = users_pb2.UserManagement()
    user_data.ParseFromString(f.read())
    f.close()

    print(user_data)
예제 #2
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def load_users(b_64=None):
    user_data = users_pb2.UserManagement()
    if b_64:
        user_data.ParseFromString(base64.b64decode(b_64))
    else:
        f = open(folder + 'database.db', 'rb')
        user_data.ParseFromString(f.read())
    return user_data
예제 #3
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def user_by_id(chat_id):
    f = open(folder + 'database.db', 'rb')
    user_data = users_pb2.UserManagement()
    user_data.ParseFromString(f.read())
    f.close()

    for user in user_data.users:
        if user.chat_id == chat_id:
            return user
예제 #4
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def user_lists():
    f = open(folder + 'database.db', 'rb')
    user_data = users_pb2.UserManagement()
    user_data.ParseFromString(f.read())
    user_lst = []
    admin_lst = []
    for user in user_data.users:
        if user.chat_id != -1:
            user_lst.append(user.chat_id)
            if user.privs == 0:
                admin_lst.append(user.chat_id)
    return user_lst, admin_lst
예제 #5
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def change_user(chat_id, fuel_kind):
    f = open(folder + 'database.db', 'rb')
    user_data = users_pb2.UserManagement()
    user_data.ParseFromString(f.read())
    f.close()

    for user in user_data.users:
        if user.chat_id == chat_id:
            user.fuel_kind = fuel_kind

    f = open(folder + 'database.db', "wb")
    f.write(user_data.SerializeToString())
    f.close()
예제 #6
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def change_fuel_price_limit(chat_id, limit):
    f = open(folder + 'database.db', 'rb')
    user_data = users_pb2.UserManagement()
    user_data.ParseFromString(f.read())
    f.close()

    for user in user_data.users:
        if user.chat_id == chat_id:
            user.price_limit = limit
            user.lowest_price = limit

    f = open(folder + 'database.db', "wb")
    f.write(user_data.SerializeToString())
    f.close()
예제 #7
0
파일: users.py 프로젝트: fabian-hk/HomeBot 
def add_user(chat_id, username=None, privs=3, longitude=conf["defaults"]["default_longitude"],
             latitude=conf["defaults"]["default_latitude"]):
    user_data = users_pb2.UserManagement()

    if os.path.isfile(folder + 'database.db'):
        f = open(folder + 'database.db', 'rb')
        user_data.ParseFromString(f.read())
        f.close()

    new_user = user_data.users.add()

    new_user.chat_id = chat_id
    if username:
        new_user.username = username
    if privs != 3:
        new_user.privs = privs

    new_user.longitude = longitude
    new_user.latitude = latitude

    f = open(folder + 'database.db', "wb")
    f.write(user_data.SerializeToString())
    f.close()