forked from Alexey/DAS_2024_1
116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
|
from flask import Flask, jsonify, request
|
|||
|
|
from uuid import uuid4
|
|||
|
|
import uuid
|
|||
|
|
import requests
|
|||
|
|
|
|||
|
|
|
|||
|
|
class User:
|
|||
|
|
def __init__(self, name, surname, uuid_: uuid):
|
|||
|
|
if uuid_ is None:
|
|||
|
|
self.uuid_: uuid = uuid4()
|
|||
|
|
else:
|
|||
|
|
self.uuid_: uuid = uuid.UUID(uuid_)
|
|||
|
|
self.name: str = name
|
|||
|
|
self.surname: str = surname
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
return {
|
|||
|
|
"uuid": self.uuid_,
|
|||
|
|
"name": self.name,
|
|||
|
|
"surname": self.surname
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def to_dict_with_messages(self, messages: list):
|
|||
|
|
return {
|
|||
|
|
"uuid": self.uuid_,
|
|||
|
|
"name": self.name,
|
|||
|
|
"surname": self.surname,
|
|||
|
|
"messages": messages
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
app = Flask(__name__)
|
|||
|
|
|
|||
|
|
users: list[User] = [
|
|||
|
|
User(name='Dr.', surname='Kino', uuid_='94b171ea-39f6-4a67-9c67-061743f67cfd'),
|
|||
|
|
User(name='Caspian', surname='Holstrom', uuid_='724a3192-70dd-4909-9b0f-c9060a4ab1bd'),
|
|||
|
|
User(name='Admin', surname='Admin', uuid_='46672ea5-3d7b-4137-a0ac-efd898ca4db6')
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
messages_url = 'http://messageService:20002/'
|
|||
|
|
|
|||
|
|
|
|||
|
|
def list_jsonify():
|
|||
|
|
return jsonify([user.to_dict() for user in users])
|
|||
|
|
|
|||
|
|
@app.route('/', methods=['GET'])
|
|||
|
|
def get_all():
|
|||
|
|
return list_jsonify(), 200
|
|||
|
|
|
|||
|
|
@app.route('/<uuid:uuid_>', methods=['GET'])
|
|||
|
|
def get_one(uuid_):
|
|||
|
|
for user in users:
|
|||
|
|
if user.uuid_ == uuid_:
|
|||
|
|
return user.to_dict(), 200
|
|||
|
|
|
|||
|
|
return f'Пользователь с uuid {uuid_} не найден', 404
|
|||
|
|
|
|||
|
|
@app.route('/info/<uuid:uuid_>', methods=['GET'])
|
|||
|
|
def get_one_with_messages(uuid_):
|
|||
|
|
for user in users:
|
|||
|
|
if user.uuid_ == uuid_:
|
|||
|
|
response = requests.get(messages_url + f'by-user/{uuid_}')
|
|||
|
|
print(response.json())
|
|||
|
|
return user.to_dict_with_messages(response.json()), 200
|
|||
|
|
|
|||
|
|
return f'Пользователь с uuid {uuid_} не найден', 404
|
|||
|
|
|
|||
|
|
@app.route('/check/<uuid:uuid_>', methods=['GET'])
|
|||
|
|
def check_exist(uuid_):
|
|||
|
|
for user in users:
|
|||
|
|
if user.uuid_ == uuid_:
|
|||
|
|
return '', 200
|
|||
|
|
return '', 404
|
|||
|
|
|
|||
|
|
@app.route('/', methods=['POST'])
|
|||
|
|
def create():
|
|||
|
|
data = request.json
|
|||
|
|
name = data.get('name', None)
|
|||
|
|
surname = data.get('surname', None)
|
|||
|
|
if name is None or surname is None:
|
|||
|
|
return 'Недостаточно информации для создания пользователя', 404
|
|||
|
|
|
|||
|
|
new_user = User(name, surname, None)
|
|||
|
|
users.append(new_user)
|
|||
|
|
return get_one(new_user.uuid_)
|
|||
|
|
|
|||
|
|
@app.route('/<uuid:uuid_>', methods=['PUT'])
|
|||
|
|
def update_by_id(uuid_):
|
|||
|
|
data = request.json
|
|||
|
|
new_name = data.get('name', None)
|
|||
|
|
new_surname = data.get('surname', None)
|
|||
|
|
|
|||
|
|
for user in users:
|
|||
|
|
if user.uuid_ == uuid_:
|
|||
|
|
if new_name is not None:
|
|||
|
|
user.name = new_name
|
|||
|
|
if new_surname is not None:
|
|||
|
|
user.surname = new_surname
|
|||
|
|
return get_one(user.uuid_)
|
|||
|
|
|
|||
|
|
return f'Пользователь с uuid {uuid_} не найден', 404
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route('/<uuid:uuid_>', methods=['DELETE'])
|
|||
|
|
def delete(uuid_):
|
|||
|
|
for user in users:
|
|||
|
|
if user.uuid_ == uuid_:
|
|||
|
|
users.remove(user)
|
|||
|
|
return 'Пользователь удален', 200
|
|||
|
|
|
|||
|
|
return f'Пользователь с uuid {uuid_} не найден', 404
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
app.run(host='0.0.0.0', port=20001, debug=True)
|