ownerrez-wrapper 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

ownerrez_wrapper/__init__.py

@@ -8,8 +8,9 @@ A Python wrapper for the OwnerRez API.
 from .api import API
 from .cli import main as cli
 
-__version__ = "0.1.0"
+__version__ = "0.2.0"
 __author__ = "Geody Moore"
-__email__ = "geody.moore@gmail.com"
+__author_email__ = "geody.moore@gmail.com"
+__description__ = "A Python wrapper for the OwnerRez API"
 
 __all__ = ["API", "cli"]

ownerrez_wrapper/api.py

@@ -48,9 +48,14 @@ class API(object):
         params = {'since_utc': since_utc, 'property_id': property_id}
         booking_list = restAdapt.get(endpoint='bookings', ep_params=params)
         
-        for booking in booking_list.data['items']:
-            booking = Booking(**booking)
-            results.append(booking)
+        if isinstance(booking_list.data, list):
+            for booking in booking_list.data:
+                booking = Booking(**booking)
+                results.append(booking)
+        elif isinstance(booking_list.data, dict) and 'items' in booking_list.data:
+            for booking in booking_list.data['items']:
+                booking = Booking(**booking)
+                results.append(booking)
         return results
     
     def getbooking(self, booking_id: int) -> Booking:
@@ -76,9 +81,11 @@ class API(object):
         today = datetime.today()
         bookings = self.getbookings(property_id=property_id, since_utc=today)
         for booking in bookings:
+            if booking.is_block:
+                continue
             # Convert string dates to datetime if they aren't already
             arrival = booking.arrival if isinstance(booking.arrival, datetime) else datetime.strptime(booking.arrival, "%Y-%m-%d")
             departure = booking.departure if isinstance(booking.departure, datetime) else datetime.strptime(booking.departure, "%Y-%m-%d")
-            if arrival <= today and departure >= today:
+            if arrival <= today and departure >= today and booking.status == "active":
                 return True
         return False

{ownerrez_wrapper-0.1.0.dist-info → ownerrez_wrapper-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ownerrez-wrapper
-Version: 0.1.0
+Version: 0.2.0
 Summary: A Python wrapper for the OwnerRez API
 Home-page: https://github.com/gmoorevt/ownerrez-wrapper
 Author: Geody Moore

ownerrez_wrapper-0.2.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+ownerrez_wrapper/__init__.py,sha256=T9VhZnbqMPykgLwmg7Ze8zKb0lVys5aKP3t83umPKck,317
+ownerrez_wrapper/api.py,sha256=CBvstz4MqTb8OPaT_fOniRb4JxYDd0XckjCCub3mouQ,3356
+ownerrez_wrapper/cli.py,sha256=44xW8FJk1YgIhqerP93hkHAWGTIWjakgRv_cpeZs7UI,4132
+ownerrez_wrapper/constants.py,sha256=Ix4x3_9aXUymp6Kcl2AzxrGM2EXHghD0-uYZLclYTVU,49
+ownerrez_wrapper/exceptions.py,sha256=HKIlPHD9A-J8aXnTglPYcxo6hx-_zQQtCgNoEgVklQI,97
+ownerrez_wrapper/model.py,sha256=Gps5BlP-0jhP7P7ludbRAuvm9g2HSiaN4cGtN76u9MM,7289
+ownerrez_wrapper/restAdapter.py,sha256=nFEs8XEOYnwGCTDiEQB4gQi5e7HTjm4Bys_NMTJGIlc,3254
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/test_api.py,sha256=RJ9suPSmL7uix_55QtLCdrbh1GrN2fb02exu78usL9w,4340
+ownerrez_wrapper-0.2.0.dist-info/LICENSE,sha256=K_mUywmexnvIytD3Tez6g71VomULBMeiG5T0vpswsuc,1068
+ownerrez_wrapper-0.2.0.dist-info/METADATA,sha256=-5uvP5UuSzzcJOX2lK_6AG4dHFqUz2Ps8K11_ZPSlcQ,4976
+ownerrez_wrapper-0.2.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ownerrez_wrapper-0.2.0.dist-info/entry_points.txt,sha256=_VDZya-q8qV4nm6vSOuaWFoX5Mp7v3ToTY-uyY-vlU4,55
+ownerrez_wrapper-0.2.0.dist-info/top_level.txt,sha256=byMzRUOQek_TqAI4k_0nj4xi9HtKKoHZ_t4XBIwUduU,23
+ownerrez_wrapper-0.2.0.dist-info/RECORD,,

tests/test_api.py

@@ -110,6 +110,10 @@ def test_isunitbooked(api, mock_booking_response, monkeypatch):
     def mock_get(*args, **kwargs):
         return Result(status=200, message="OK", data=mock_booking_response)
     
+    # Update mock response to include status and is_block fields
+    mock_booking_response['items'][0]['status'] = 'active'
+    mock_booking_response['items'][0]['is_block'] = False
+    
     monkeypatch.setattr("ownerrez_wrapper.restAdapter.RestAdapter.get", mock_get)
     is_booked = api.isunitbooked(property_id=1)
     assert is_booked is True
@@ -119,6 +123,10 @@ def test_isunitbooked_not_booked(api, mock_booking_response, monkeypatch):
     def mock_get(*args, **kwargs):
         return Result(status=200, message="OK", data=mock_booking_response)
     
+    # Update mock response to include status and is_block fields
+    mock_booking_response['items'][0]['status'] = 'active'
+    mock_booking_response['items'][0]['is_block'] = False
+    
     monkeypatch.setattr("ownerrez_wrapper.restAdapter.RestAdapter.get", mock_get)
     is_booked = api.isunitbooked(property_id=1)
     assert is_booked is False

ownerrez_wrapper-0.1.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-ownerrez_wrapper/__init__.py,sha256=S4PQiSD4uqBXFDlnQZ6tLDRX_KHH8EAtVZjDZSecubM,252
-ownerrez_wrapper/api.py,sha256=sD_hlZ5Slb7cprI6os4nVdPeQSYsqteuDQ-DWFw70N4,2993
-ownerrez_wrapper/cli.py,sha256=44xW8FJk1YgIhqerP93hkHAWGTIWjakgRv_cpeZs7UI,4132
-ownerrez_wrapper/constants.py,sha256=Ix4x3_9aXUymp6Kcl2AzxrGM2EXHghD0-uYZLclYTVU,49
-ownerrez_wrapper/exceptions.py,sha256=HKIlPHD9A-J8aXnTglPYcxo6hx-_zQQtCgNoEgVklQI,97
-ownerrez_wrapper/model.py,sha256=Gps5BlP-0jhP7P7ludbRAuvm9g2HSiaN4cGtN76u9MM,7289
-ownerrez_wrapper/restAdapter.py,sha256=nFEs8XEOYnwGCTDiEQB4gQi5e7HTjm4Bys_NMTJGIlc,3254
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_api.py,sha256=g8POEEiBjh6rxWiIJTbgv0kc7c7kSBvyEbHhUsNk80c,3966
-ownerrez_wrapper-0.1.0.dist-info/LICENSE,sha256=K_mUywmexnvIytD3Tez6g71VomULBMeiG5T0vpswsuc,1068
-ownerrez_wrapper-0.1.0.dist-info/METADATA,sha256=1ZDJmddxB976zPQqGScUaV65-edJVN0kHk5p-sF5wKw,4976
-ownerrez_wrapper-0.1.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ownerrez_wrapper-0.1.0.dist-info/entry_points.txt,sha256=_VDZya-q8qV4nm6vSOuaWFoX5Mp7v3ToTY-uyY-vlU4,55
-ownerrez_wrapper-0.1.0.dist-info/top_level.txt,sha256=byMzRUOQek_TqAI4k_0nj4xi9HtKKoHZ_t4XBIwUduU,23
-ownerrez_wrapper-0.1.0.dist-info/RECORD,,