Вот, но это АЗЫ для твой проги
from json import load, dump
class Users:
"""Общий класс для работы с пользователями в формате JSON"""
def __init__(self, filename='users.json'):
self.filename = filename
self.load()
def load(self):
"""Загрузить файл с пользователями"""
try:
with open(self.filename) as fp:
self.users = load(fp)
except FileNotFoundError:
self.users = {}
def save(self):
"""Сохраняет файл с пользователями"""
with open(self.filename, 'w') as fp:
dump(self.users, fp)
def add_user(self, name, lakes):
"""Добавляет пользователя"""
user_id = str(len(self.users) + 1)
self.users[user_id] = {'name': name, 'lakes': lakes} # Здесь можно добавить поля
self.save()
def remove_user(self, user_id):
"""Удаляет пользователя"""
if user_id in self.users:
del self.users[user_id]
self.save()
def get_users(self):
"""Получить пользователей"""
return self.users
def get_user(self, user_id):
"""Получить пользователя под номером..."""
return self.users.get(user_id, None)
# Иниц-я
users = Users()
print(users.users)
# добавлять пользователя
users.add_user("Poll", 124)
# Сохранить файл
users.save()
