54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from models import db, Event
|
|
from datetime import datetime
|
|
|
|
class EventService:
|
|
@staticmethod
|
|
def create_event(data):
|
|
print("#########")
|
|
print(data)
|
|
new_event = Event(
|
|
title=data['title'],
|
|
description=data.get('description', ''),
|
|
location=data.get('location', ''),
|
|
duedate=datetime.strptime(data['duedate'], '%Y-%m-%dT%H:%M:%S'),
|
|
user_id=data['user_id']
|
|
)
|
|
db.session.add(new_event)
|
|
db.session.commit()
|
|
return new_event
|
|
|
|
@staticmethod
|
|
def get_all_user_events(user_id):
|
|
events=Event.query.filter_by(user_id=user_id, deleted=False).all()
|
|
if(events):
|
|
return [event.to_dict() for event in events]
|
|
else:
|
|
return []
|
|
|
|
@staticmethod
|
|
def get_event_by_id(event_id, user_id):
|
|
event = Event.query.filter_by(id=event_id,user_id=user_id, deleted=False).first()
|
|
if(event):
|
|
return event.to_dict()
|
|
else:
|
|
return []
|
|
|
|
@staticmethod
|
|
def update_event(event_id, data):
|
|
event = Event.query.get(event_id)
|
|
event.title = data['title']
|
|
event.description = data.get('description', '')
|
|
event.location = data.get('location', '')
|
|
event.time = datetime.strptime(data['duedate'], '%Y-%m-%dT%H:%M:%S')
|
|
db.session.commit()
|
|
return event
|
|
|
|
@staticmethod
|
|
def delete_event(event_id):
|
|
event = Event.query.filter_by(id=event_id, deleted=False).first()
|
|
if event:
|
|
event.deleted = True
|
|
db.session.commit()
|
|
else:
|
|
return None
|
|
return event
|