updated some NoSQL db examples in class

Author: PythonCHB

examples/nosql/address_book_model.py

@@ -108,7 +108,7 @@ class Business(Household):
     pass
 
 
-class AddressBook(object):
+class AddressBook:
     """
     And address book -- has people, households, businesses.
 

examples/nosql/address_book_mongo.py

@@ -168,7 +168,7 @@ class Business(Household):
     pass
 
 
-class AddressBook(object):
+class AddressBook:
     """
     An address book -- has people, households, businesses.
 
@@ -199,13 +199,13 @@ def __init__(self,
         self.households = db.households
 
     def add_person(self, person):
-        self.people.insert(person.to_dict())
+        self.people.insert_one(person.to_dict())
 
     def add_household(self, household):
-        self.households.insert(household.to_dict())
+        self.households.insert_one(household.to_dict())
 
     def add_business(self, business):
-        self.businesses.insert(business.to_dict())
+        self.businesses.insert_one(business.to_dict())
 
     def __str__(self):
         msg = ["An Address Book:"]
@@ -341,7 +341,7 @@ def create_sample():
                            )
 
 
-    address_book = AddressBook()
+    address_book = AddressBook(fresh=True)
 
     address_book.add_person(chris)
     address_book.add_person(donna)

examples/nosql/address_book_zodb.py

@@ -10,6 +10,7 @@
 import persistent  # from ZODB
 from persistent.list import PersistentList
 
+
 class Person(persistent.Persistent):
     """
     class to represent an individual person
@@ -69,7 +70,8 @@ def __init__(self,
         self.zip_code = str(zip_code).strip()
 
     def __str__(self):
-        msg = "{line_1}\n{line_2}\n{city} {state} {zip_code}\n".format(**self.__dict__)
+        msg = "an address"
+        #msg = "{line_1}\n{line_2}\n{city} {state} {zip_code}\n".format()
         return msg
 
 

examples/nosql/test_address_book_model.py

@@ -6,15 +6,21 @@
 $ mongod --dbpath=mongo_data/
 """
 
-import address_book_model as model
+import pytest
+
+import ZODB
+
+# import address_book_model as model
 # import address_book_zodb as model
-# import address_book_mongo as model
+import address_book_mongo as model
 
 
-a_book = model.create_sample()
+@pytest.fixture
+def a_book():
+    return model.create_sample()
 
 
-def test_name_search():
+def test_name_search(a_book):
     """find a single person by first name"""
 
     people = a_book.find_people('chris')
@@ -24,7 +30,7 @@ def test_name_search():
     assert people[0].last_name == 'Barker'
 
 
-def test_name_search2():
+def test_name_search2(a_book):
     people = a_book.find_people('barKer')
     first_names = [p.first_name for p in people]
 
@@ -33,16 +39,45 @@ def test_name_search2():
     assert 'Donna' in first_names
 
 
-def test_zip_search():
+def test_zip_search(a_book):
     locations = a_book.find_zip_codes(98105)
 
     assert len(locations) == 1
     assert locations[0].name == 'Python Certification Program'
 
 
-def test_state_search():
+def test_state_search(a_book):
     locations = a_book.find_state('WA')
     names = [l.name for l in locations]
 
     assert "The Barkers" in names
     assert "Python Certification Program" in names
+
+
+# def test_household():
+#     house = model.household()
+
+
+# def test_add_person_to_household(a_book):
+#     sassy = model.Person(first_name="Sassy",
+#                    last_name="Barker"
+#                    )
+
+#     print(a_book.households)
+#     house = a_book.locations[0]
+#     house.people.append(sassy)
+
+
+#     db = ZODB.DB('address_book_zodb.fs')
+#     connection = db.open()
+#     root = connection.root
+#     a_book = root.address_book
+
+#     house = a_book.households[0]
+#     print(house)
+
+#     assert False
+
+
+
+