class UpdateTask(Resource):
@jwt_required
@marshal_with(TaskModel.get_marshaller())
def put(self, task_id):
parser = reqparse.RequestParser()
parser.add_argument('time_next_instance', type=moment, required=False)
parser.add_argument('time_interval', type=int, required=False)
parser.add_argument('name', type=str)
parser.add_argument('description', type=str)
parser.add_argument('km_interval', type=int, required=False)
parser.add_argument('km_next_instance', type=float, required=False)
data = parser.parse_args()
task: TaskModel = TaskModel.find_by_id(task_id)
if not task:
abort(404, message=TASK_DOESNT_EXIST)
if not task.is_reocurrent:
abort(401, message=NON_REOCURRENT_TASKS_CANNOT_BE_UPDATED)
community_member_ids = [m.id for m in task.community.users]
user = UserModel.find_by_username(get_jwt_identity())
if user.id not in community_member_ids:
abort(401, message=UNAUTHORIZED)
if not (data['time_next_instance'] and data['time_interval'] or data['km_interval'] and data[
'km_next_instance']) or data['km_interval'] and (data['time_interval'] or data['time_next_instance']) or \
data['time_interval'] and (data['km_interval'] or data['km_next_instance']):
abort(400, message=TASK_MUST_BE_EITHER_TIME_OR_KM_TRIGGERED)
time_interval = None
if data['time_interval']:
time_interval = timedelta(days=data['time_interval'])
task.time_interval = time_interval
task.time_next_instance = data['time_next_instance']
task.km_interval = data['km_interval']
task.km_next_instance = data['km_next_instance']
task.name = data['name']
task.description = data['description']
try:
task.persist()
set_km_to_next_instance(task)
return task, 200
except:
abort(500, message=INTERNAL_SERVER_ERROR)
class GetCommunityTasks(Resource):
@jwt_required
@marshal_with(TaskModel.get_marshaller())
def get(self, community_id):
tasks: List[TaskModel] = TaskModel.find_by_community(community_id)
community: CommunityModel = CommunityModel.find_by_id(community_id)
community_member_ids = [m.id for m in community.users]
user = UserModel.find_by_username(get_jwt_identity())
if user.id not in community_member_ids:
abort(401, message=UNAUTHORIZED)
set_km_to_next_instance(tasks)
return tasks, 200
def get_marshaller():
return {
'id':
fields.Integer,
'time_created':
fields.DateTime,
'time_updated':
fields.DateTime,
'time_finished':
fields.DateTime,
'task':
fields.Nested(TaskModel.get_marshaller()),
'is_open':
fields.Boolean,
'km_created_at':
fields.Float,
'finished_by':
fields.Nested(UserModel.get_marshaller(), allow_null=True)
}
class GetTask(Resource):
@jwt_required
@marshal_with(TaskModel.get_marshaller())
def get(self, task_id):
task: TaskModel = TaskModel.find_by_id(task_id)
if not task:
abort(404, message=TASK_DOESNT_EXIST)
community_member_ids = [m.id for m in task.community.users]
user = UserModel.find_by_username(get_jwt_identity())
if user.id not in community_member_ids:
abort(401, message=UNAUTHORIZED)
set_km_to_next_instance(task)
return task, 200
class CreateTask(Resource):
@jwt_required
@marshal_with(TaskModel.get_marshaller())
def post(self, community_id):
parser = reqparse.RequestParser()
parser.add_argument('time_next_instance', type=moment, required=False)
parser.add_argument('time_interval', type=int, required=False)
parser.add_argument('name', type=str)
parser.add_argument('description', type=str)
parser.add_argument('km_interval', type=int, required=False)
parser.add_argument('km_next_instance', type=float, required=False)
parser.add_argument('is_reocurrent', type=bool, required=True)
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 data['is_reocurrent'] and (
not (data['time_next_instance'] and data['time_interval'] or data['km_interval'] and data[
'km_next_instance']) or data['km_interval'] and (data['time_interval'] or data['time_next_instance']) or \
data['time_interval'] and (data['km_interval'] or data['km_next_instance'])):
abort(400, message=TASK_MUST_BE_EITHER_TIME_OR_KM_TRIGGERED)
newest_tour: TourModel = TourModel.find_newest_tour_for_community(community_id)
if data['km_next_instance'] and data['km_next_instance'] < newest_tour.end_km:
abort(400, message=TASK_KM_NEXT_INSTANCE_MUST_BE_HIGHER_THEN_CURRENT_KM)
if data['time_next_instance'] and data['time_next_instance'] < datetime.now(pytz.timezone('Europe/Berlin')):
abort(400, message=TASK_TIME_NEXT_INSTANCE_MUST_BE_HIGHER_THEN_CURRENT_TIME)
time_interval = None
if data['time_interval']:
time_interval = timedelta(days=data['time_interval'])
new_task = TaskModel(
owner=owner,
community=community,
time_interval=time_interval,
time_next_instance=data['time_next_instance'],
km_interval=data['km_interval'],
km_next_instance=data['km_next_instance'],
name=data['name'],
description=data['description'],
is_reocurrent=data['is_reocurrent']
)
try:
new_task.persist()
set_km_to_next_instance(new_task)
if not data['is_reocurrent']:
new_task_instance = TaskInstanceModel(
task=new_task,
is_open=True,
community=new_task.community
)
new_task_instance.persist()
return new_task, 201
except:
abort(500, message=INTERNAL_SERVER_ERROR)