class RunningUserTours(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def get(self):
user = UserModel.find_by_username(get_jwt_identity())
return TourModel.find_running_by_user(user.id), 200
class RunningCommunityTours(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def get(self, community_id):
user = UserModel.find_by_username(get_jwt_identity())
community = CommunityModel.find_by_id(community_id)
if not community:
abort(400, message=COMMUNIY_DOESNT_EXIST)
if user.id not in [u.id for u in community.users]:
abort(401, message=UNAUTHORIZED)
return TourModel.find_running_by_community(community_id), 200
class ForceFinishTour(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def put(self, community_id, id):
data = finish_tour_parser.parse_args()
tour = TourModel.find_by_id(id)
user = UserModel.find_by_username(get_jwt_identity())
if not tour:
abort(404, message=TOUR_NOT_FOUND)
if user.id == tour.owner.id:
abort(401, message=UNAUTHORIZED)
if community_id != tour.community.id:
abort(401, message=UNAUTHORIZED)
if user.id not in [u.id for u in tour.community.users]:
abort(401, message=UNAUTHORIZED)
if tour.end_km:
abort(400, message=TOUR_HAS_ALREADY_BEEN_FINISHED)
if not data['passengers']:
data['passengers'] = []
passengers_changed = np.setdiff1d(data['passengers'], [p.id for p in tour.passengers]) > 0
if len(data['passengers']) != len(tour.passengers) or passengers_changed:
abort(400, message=PASSENGER_LIST_CANNOT_BE_CHANGED_WHEN_FORCE_FINISHING_A_TOUR)
tour.end_time = datetime.datetime.now(pytz.utc)
tour.end_km = data['end_km']
tour.comment = data['comment']
tour.parking_position = data['parking_position']
tour.is_force_finished = True
tour.force_finished_by = user
tour.persist()
create_km_triggered_task_instances(community_id, tour.end_km)
return tour, 200
class LatestTour(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def get(self, community_id):
user = UserModel.find_by_username(get_jwt_identity())
community = CommunityModel.find_by_id(community_id)
if not community:
abort(400, message=COMMUNIY_DOESNT_EXIST)
if user.id not in [u.id for u in community.users]:
abort(401, message=UNAUTHORIZED)
tour = TourModel.find_newest_tour_for_community(community_id)
if not tour:
abort(400, message=NO_TOUR_EXISTING)
return tour, 200
class AllTours(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def post(self, community_id):
data = parser.parse_args()
owner = UserModel.find_by_username(get_jwt_identity())
community: CommunityModel = CommunityModel.find_by_id(community_id)
community_member_ids = [m.id for m in community.users]
if not community:
abort(400, message=COMMUNIY_DOESNT_EXIST)
if owner.id not in community_member_ids:
abort(401, message=UNAUTHORIZED)
if TourModel.find_running_by_community(community_id):
abort(400, message=CANT_START_TOUR_WHEN_HAVING_UNFINISHED_TOURS_IN_COMMUNITY)
passengers = []
if data['passengers']:
for passenger_id in set(data['passengers']):
if passenger_id not in community_member_ids:
abort(400, message=PASSENGERS_MUST_BE_COMMUNITY_MEMBERS)
else:
passengers.append([u for u in community.users if u.id == passenger_id][0])
new_tour = TourModel(
owner=owner,
community=community,
start_time=datetime.datetime.now(pytz.utc),
start_km=data['start_km'],
passengers=passengers
)
try:
new_tour.persist()
return new_tour, 201
except:
abort(500, message=INTERNAL_SERVER_ERROR)
class FinishTour(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def put(self, community_id, id):
data = finish_tour_parser.parse_args()
tour = TourModel.find_by_id(id)
user = UserModel.find_by_username(get_jwt_identity())
community: CommunityModel = CommunityModel.find_by_id(community_id)
community_member_ids = [m.id for m in community.users]
if not tour:
abort(404, message=TOUR_NOT_FOUND)
if not user.id == tour.owner.id:
abort(401, message=UNAUTHORIZED)
if tour.end_km:
abort(400, message=TOUR_HAS_ALREADY_BEEN_FINISHED)
passengers = []
if data['passengers']:
for passenger_id in set(data['passengers']):
if passenger_id not in community_member_ids:
abort(400, message=PASSENGERS_MUST_BE_COMMUNITY_MEMBERS)
else:
passengers.append([u for u in community.users if u.id == passenger_id][0])
tour.end_time = datetime.datetime.now(pytz.utc)
tour.passengers = passengers
tour.end_km = data['end_km']
tour.comment = data['comment']
tour.parking_position = data['parking_position']
tour.persist()
create_km_triggered_task_instances(community_id, tour.end_km)
return tour, 200
class SingleTour(Resource):
@jwt_required
@marshal_with(TourModel.get_marshaller())
def put(self, community_id, id):
data = edit_tour_parser.parse_args()
tour = TourModel.find_by_id(id)
user = UserModel.find_by_username(get_jwt_identity())
community: CommunityModel = CommunityModel.find_by_id(community_id)
community_member_ids = [m.id for m in community.users]
if not tour:
abort(404, message=TOUR_NOT_FOUND)
if not user.id == tour.owner.id:
abort(401, message=UNAUTHORIZED)
passengers = []
if data['passengers']:
for passenger_id in set(data['passengers']):
if passenger_id not in community_member_ids:
abort(400, message=PASSENGERS_MUST_BE_COMMUNITY_MEMBERS)
else:
passengers.append([u for u in community.users if u.id == passenger_id][0])
tour.comment = data['comment']
tour.parking_position = data['parking_position']
if not tour.is_open:
abort(400, message=CANNOT_UPDATE_SENSITIVE_TOUR_DATA_WHEN_TOUR_IS_ALREADY_PAYED_FOR)
else:
if data['end_km'] <= data['start_km']:
abort(400, message=END_KM_MUST_BE_GREATER_START_KM)
tour.end_km = data['end_km']
tour.start_km = data['start_km']
tour.passengers = passengers
tour.persist()
return tour, 200
@jwt_required
@marshal_with(SimpleMessage.get_marshaller())
def delete(self, community_id, id):
tour = TourModel.find_by_id(id)
user = UserModel.find_by_username(get_jwt_identity())
if not tour:
abort(404, message=TOUR_NOT_FOUND)
if not user.id == tour.owner.id:
abort(401, message=UNAUTHORIZED)
try:
TourModel.delete_by_id(id)
except NoData:
abort(404, message=TOUR_NOT_FOUND)
return SimpleMessage(TOUR_DELETED), 200
@jwt_required
@marshal_with(TourModel.get_marshaller())
def get(self, community_id, id):
tour = TourModel.find_by_id(id)
user = UserModel.find_by_username(get_jwt_identity())
if not tour:
abort(404, message=TOUR_NOT_FOUND)
if user.id not in [u.id for u in tour.community.users]:
abort(401, message=UNAUTHORIZED)
return tour, 200
