def send_message_code(user, business_id, message, user_id):
recv_user = None
if user_id:
if not business.user_owns_business(user.user_id(), business_id):
return {"error": "You don't own that business"}
recv_user = CoUser.get(user_id)
if not recv_user:
return {"error": "no such user"}
interaction = InteractionRecord.gql("WHERE business_id = :1 AND user = :2", business_id, recv_user).get()
if not interaction:
return {"error": "no reason to talk to that person"}
contractor = True
else:
interaction = InteractionRecord.guaranteed_get(business_id, user)
contractor = False
senderoo = Solicitation(
interaction = interaction,
from_contractor = contractor,
when = js_time(),
message = message
)
senderoo.put()
send_emails.message_recieved(user, business_id,recv_user, message)
return model_to_dict(senderoo)
def add_booking_code(self, user, business_id, start, end, user_with):
if user_with:
# Make sure they are allowed to propose to this user
if not business.user_owns_business(user.user_id(), business_id):
return {"error": "not your business bro"}
user_with = CoUser.get(user_with)
if not user_with:
return {"error":"that guy does not exist"}
interaction = InteractionRecord.gql("WHERE business_id = :1 AND user = :2", business_id, user_with).get()
if not interaction:
return {"error":"You have no reason to talk to that user"}
user = user_with
contractor = True
client = False
else:
contractor = False
client = True
need_to_create_this = InteractionRecord.guaranteed_get(business_id, user)
new_booking = Booking(
user = user,
business_id = business_id,
start_time = start,
end_time = end,
approved_by_customer = client,
approved_by_contractor = contractor
)
new_booking.put()
send_emails.booking_created(user, business_id, start, end, user_with)
return new_booking
def get_client_messages(self, user, business_id, client_id):
if not user or not business.user_owns_business(user.user_id(), business_id):
return {"error": "permissions"}
interaction = InteractionRecord.gql("WHERE business_id = :1 AND user = :2", business_id, Key(client_id)).get()
if not interaction:
return []
messages = Solicitation.gql("WHERE interaction = :1 ORDER BY when", interaction)
return gql_to_raw(messages)
def delete_booking(self, user, booking_id):
booking = Booking.get(booking_id)
if not booking:
return {"error": "No such booking"}
if str(booking.user.key()) == str(user.key()):
booking.delete()
return {"success": True}
elif business.user_owns_business(user.user_id(), booking.business_id):
booking.delete()
return {"success": True}
else:
return {"error": "insufficient permissions"}
def approve_booking(self, user, booking_id):
booking = Booking.get(booking_id)
if not booking:
return {"error": "No such booking"}
if str(booking.user.key()) == str(user.key()):
booking.approved_by_customer = True
booking.put()
return model_to_dict(booking)
if business.user_owns_business(user.user_id(), booking.business_id):
booking.approved_by_contractor = True
booking.put()
return model_to_dict(booking)
return {"error": "Insufficient permissions"}
def get_messages(self, user, business_id, user_id = None):
if user_id:
if not business.user_owns_business(user.user_id(), business_id):
return {"error": "You do not own that business"}
user = CoUser.get(user_id)
if not user:
return {"error": "What are you doing?"}
interaction = InteractionRecord.gql("WHERE business_id = :1 AND user = :2", business_id, user).get()
if not interaction:
return []
messages = Solicitation.gql("WHERE interaction = :1 ORDER BY when", interaction)
return gql_to_raw(messages)
def get_clients(self, user, business_id):
# This is the fun one
# Going to need some caching
user_id = user.user_id()
if not user or not business.user_owns_business(user.user_id(), business_id):
return {"error": "not allowed"}
the_real_deal = InteractionRecord.gql("WHERE business_id = :1", business_id)
ret = []
# This is so that all of the user's data comes through
for i in the_real_deal:
ret.append(model_to_dict(i.user))
return ret
def get_bookings(self, user, business_id, user_id=None):
# My intuition tells me that this is bad design
# My laziness told it to shut up
if user_id:
if not business.user_owns_business(user.user_id(), business_id):
return {"error": "You don't own that business"}
# At this point, permissions are good
user = CoUser.get(user_id)
if not user:
return []
bookings = Booking.gql("WHERE user = :1 AND business_id = :2", user, business_id)
return gql_to_raw(bookings)
def respond(self, user, business_id, user_to, message):
""" Used by businesses to respond to user queries
"""
if not business.user_owns_business(user.user_id(), business_id):
return {"error": "NOT ALLOWED DOG"}
interaction = InteractionRecord.get(business_id, user_to)
if not interaction:
return {"error": "You cannot contact that user"}
senderoo = Solicitation(
interaction = interaction,
from_contractor=True,
when = js_time(),
message = message
)
senderoo.put()
# TODO: tell everyone again
return model_to_dict(senderoo)