Merge branch 'master' of https://github.com/mikeckennedy/mongodb-quickstart-course

* 'master' of https://github.com/mikeckennedy/mongodb-quickstart-course: final code. guest can book a snake into a cage Register and list cages Create account implemented. Model classes converted to MongoEngine entities Code from "register connections". Added plain python classes for modeling. misspelling.

Author: mikeckennedy

src/snake_bnb/reqjuirements.txt renamed to src/snake_bnb/requirements.txt

@@ -0,0 +1,18 @@
+import mongoengine
+
+
+class Booking(mongoengine.EmbeddedDocument):
+ guest_owner_id = mongoengine.ObjectIdField()
+ guest_snake_id = mongoengine.ObjectIdField()
+
+ booked_date = mongoengine.DateTimeField()
+ check_in_date = mongoengine.DateTimeField(required=True)
+ check_out_date = mongoengine.DateTimeField(required=True)
+
+ review = mongoengine.StringField()
+ rating = mongoengine.IntField(default=0)
+
+ @property
+ def duration_in_days(self):
+ dt = self.check_out_date - self.check_in_date
+ return dt.days

src/snake_bnb/src/data/bookings.py

@@ -0,0 +1,22 @@
+import datetime
+import mongoengine
+
+from data.bookings import Booking
+
+
+class Cage(mongoengine.Document):
+ registered_date = mongoengine.DateTimeField(default=datetime.datetime.now)
+
+ name = mongoengine.StringField(required=True)
+ price = mongoengine.FloatField(required=True)
+ square_meters = mongoengine.FloatField(required=True)
+ is_carpeted = mongoengine.BooleanField(required=True)
+ has_toys = mongoengine.BooleanField(required=True)
+ allow_dangerous_snakes = mongoengine.BooleanField(default=False)
+
+ bookings = mongoengine.EmbeddedDocumentListField(Booking)
+
+ meta = {
+ 'db_alias': 'core',
+ 'collection': 'cages'
+ }

src/snake_bnb/src/data/cages.py

@@ -0,0 +1,5 @@
+import mongoengine
+
+
+def global_init():
+ mongoengine.register_connection(alias='core', name='snake_bnb')

src/snake_bnb/src/data/mongo_setup.py

@@ -0,0 +1,16 @@
+import datetime
+import mongoengine
+
+
+class Owner(mongoengine.Document):
+ registered_date = mongoengine.DateTimeField(default=datetime.datetime.now)
+ name = mongoengine.StringField(required=True)
+ email = mongoengine.StringField(required=True)
+
+ snake_ids = mongoengine.ListField()
+ cage_ids = mongoengine.ListField()
+
+ meta = {
+ 'db_alias': 'core',
+ 'collection': 'owners'
+ }

src/snake_bnb/src/data/owners.py

@@ -0,0 +1,16 @@
+import datetime
+import mongoengine
+
+
+class Snake(mongoengine.Document):
+ registered_date = mongoengine.DateTimeField(default=datetime.datetime.now)
+ species = mongoengine.StringField(required=True)
+
+ length = mongoengine.FloatField(required=True)
+ name = mongoengine.StringField(required=True)
+ is_venomous = mongoengine.BooleanField(required=True)
+
+ meta = {
+ 'db_alias': 'core',
+ 'collection': 'snakes'
+ }

src/snake_bnb/src/data/snakes.py

@@ -1,10 +1,12 @@
-active_account = None
+from data.owners import Owner
+import servies.data_service as svc
+
+active_account: Owner = None
 
 
 def reload_account():
  global active_account
  if not active_account:
  return
 
- # TODO: pull owner account from the database.
- pass
+ active_account = svc.find_account_by_email(active_account.email)

src/snake_bnb/src/infrastructure/state.py

@@ -1,10 +1,11 @@
 from colorama import Fore
 import program_guests
 import program_hosts
+import data.mongo_setup as mongo_setup
 
 
 def main():
- # TODO: Setup mongoengine global values
+ mongo_setup.global_init()
 
  print_header()
 

src/snake_bnb/src/program.py

@@ -1,5 +1,9 @@
+import datetime
+from dateutil import parser
+
 from infrastructure.switchlang import switch
 import program_hosts as hosts
+import servies.data_service as svc
 from program_hosts import success_msg, error_msg
 import infrastructure.state as state
 
@@ -54,36 +58,114 @@ def show_commands():
 
 def add_a_snake():
  print(' ****************** Add a snake **************** ')
- # TODO: Require an account
- # TODO: Get snake info from user
- # TODO: Create the snake in the DB.
+ if not state.active_account:
+ error_msg("You must log in first to add a snake")
+ return
+
+ name = input("What is your snake's name? ")
+ if not name:
+ error_msg('cancelled')
+ return
+
+ length = float(input('How long is your snake (in meters)? '))
+ species = input("Species? ")
+ is_venomous = input("Is your snake venomous [y]es, [n]o? ").lower().startswith('y')
 
- print(" -------- NOT IMPLEMENTED -------- ")
+ snake = svc.add_snake(state.active_account, name, length, species, is_venomous)
+ state.reload_account()
+ success_msg('Created {} with id {}'.format(snake.name, snake.id))
 
 
 def view_your_snakes():
  print(' ****************** Your snakes **************** ')
+ if not state.active_account:
+ error_msg("You must log in first to view your snakes")
+ return
 
- # TODO: Require an account
- # TODO: Get snakes from DB, show details list
-
- print(" -------- NOT IMPLEMENTED -------- ")
+ snakes = svc.get_snakes_for_user(state.active_account.id)
+ print("You have {} snakes.".format(len(snakes)))
+ for s in snakes:
+ print(" * {} is a {} that is {}m long and is {}venomous.".format(
+ s.name,
+ s.species,
+ s.length,
+ '' if s.is_venomous else 'not '
+ ))
 
 
 def book_a_cage():
  print(' ****************** Book a cage **************** ')
- # TODO: Require an account
- # TODO: Verify they have a snake
- # TODO: Get dates and select snake
- # TODO: Find cages available across date range
- # TODO: Let user select cage to book.
+ if not state.active_account:
+ error_msg("You must log in first to book a cage")
+ return
+
+ snakes = svc.get_snakes_for_user(state.active_account.id)
+ if not snakes:
+ error_msg('You must first [a]dd a snake before you can book a cage.')
+ return
+
+ print("Let's start by finding available cages.")
+ start_text = input("Check-in date [yyyy-mm-dd]: ")
+ if not start_text:
+ error_msg('cancelled')
+ return
+
+ checkin = parser.parse(
+ start_text
+ )
+ checkout = parser.parse(
+ input("Check-out date [yyyy-mm-dd]: ")
+ )
+ if checkin >= checkout:
+ error_msg('Check in must be before check out')
+ return
+
+ print()
+ for idx, s in enumerate(snakes):
+ print('{}. {} (length: {}, venomous: {})'.format(
+ idx + 1,
+ s.name,
+ s.length,
+ 'yes' if s.is_venomous else 'no'
+ ))
 
- print(" -------- NOT IMPLEMENTED -------- ")
+ snake = snakes[int(input('Which snake do you want to book (number)')) - 1]
+
+ cages = svc.get_available_cages(checkin, checkout, snake)
+
+ print("There are {} cages available in that time.".format(len(cages)))
+ for idx, c in enumerate(cages):
+ print(" {}. {} with {}m carpeted: {}, has toys: {}.".format(
+ idx + 1,
+ c.name,
+ c.square_meters,
+ 'yes' if c.is_carpeted else 'no',
+ 'yes' if c.has_toys else 'no'))
+
+ if not cages:
+ error_msg("Sorry, no cages are available for that date.")
+ return
+
+ cage = cages[int(input('Which cage do you want to book (number)')) - 1]
+ svc.book_cage(state.active_account, snake, cage, checkin, checkout)
+
+ success_msg('Successfully booked {} for {} at ${}/night.'.format(cage.name, snake.name, cage.price))
 
 
 def view_bookings():
  print(' ****************** Your bookings **************** ')
- # TODO: Require an account
- # TODO: List booking info along with snake info
-
- print(" -------- NOT IMPLEMENTED -------- ")
+ if not state.active_account:
+ error_msg("You must log in first to register a cage")
+ return
+
+ snakes = {s.id: s for s in svc.get_snakes_for_user(state.active_account.id)}
+ bookings = svc.get_bookings_for_user(state.active_account.email)
+
+ print("You have {} bookings.".format(len(bookings)))
+ for b in bookings:
+ print(' * Snake: {} is booked at {} from {} for {} days.'.format(
+ snakes.get(b.guest_snake_id).name,
+ b.cage.name,
+ datetime.date(b.check_in_date.year, b.check_in_date.month, b.check_in_date.day),
+ (b.check_out_date - b.check_in_date).days
+ ))