Spaces:
Sleeping
Sleeping
File size: 1,838 Bytes
9d3162f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import json
import unittest
import requests
import os
import socket
class CaesarAIHotelBookingTest(unittest.TestCase):
def hotel_bookings_test(self):
if "Bookings" not in os.listdir():
os.mkdir("Bookings")
city = "Alicante"
price_range = 2000
booking_json= {
"city":city,
"checkin_date":"2023-8-15",
"checkout_date":"2023-8-22",
"purpose":"work",
"num_of_adults":8,
"num_of_rooms":5,
"num_of_children":0,
"price_range":price_range,
"num_of_pages":10,
"exclude_whole":"true"
}
#full_bookings = requests.post("https://caesaraiapi.onrender.com/caesaraihotelbookings",json=booking_json).json()
#print(full_bookings)
with open(f"Bookings/{city.lower()}_bookings.json","r") as f:
full_bookings = json.load(f)
#.json()
print(full_bookings)
with open(f"Bookings/{city.lower()}_bookings_lower_than_{price_range}.txt","w+") as f:
for booking in full_bookings[f"{city.lower()}_bookings_lower_than_{price_range}"]:
for key,value in booking.items():
try:
if key == "address":
key = key.capitalize()
f.write(f"{key} - {value}\n")
except KeyError as kex:
continue
f.write("\n")
#def web_socket_test(self):
# import json
# from websocket import create_connection
# ws = create_connection("ws://caesaraiapi.onrender.com/echo")
# ws.send(json.dumps({"op":"addr_sub", "addr":"dogecoin_address"}))
# result = ws.recv()
# print (result)
# ws.close()
if __name__ == "__main__":
unittest.main() |