Python DBStorageの例

コード例 #1

ファイル: test_amenity.py プロジェクト: Hillmonkey/AirBnB_clone_v3

    def setUp(self):
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session = db_storage.DBStorage()
            self.session.reload()

        self.amenity = Amenity(name="kitchen")
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session.new(self.amenity)
            self.session.save()

コード例 #2

ファイル: test_state.py プロジェクト: Hillmonkey/AirBnB_clone_v3

    def setUp(self):
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session = db_storage.DBStorage()
            self.session.reload()

        self.state = State(name="A State")
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session.new(self.state)
            self.session.save()

コード例 #3

    def setUp(self):
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session = db_storage.DBStorage()
            self.session.reload()

        self.state = State(name="Another State")
        self.city = City(name="Another City", state_id=self.state.id)

        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session.new(self.state)
            self.session.new(self.city)
            self.session.save()

コード例 #4

ファイル: test_user.py プロジェクト: Hillmonkey/AirBnB_clone_v3

    def setUp(self):
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session = db_storage.DBStorage()
            self.session.reload()

        self.state = State(name="A State")
        self.city = City(name="A City", state_id=self.state.id)
        self.user = User(email="*****@*****.**", password="******",
                         first_name="", last_name="")

        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session.new(self.state)
            self.session.new(self.city)
            self.session.new(self.user)
            self.session.save()

コード例 #5

ファイル: test_place.py プロジェクト: Hillmonkey/AirBnB_clone_v3

    def setUp(self):
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session = db_storage.DBStorage()
            self.session.reload()

        self.state = State(name="A State")
        self.city = City(name="A City", state_id=self.state.id)
        self.user = User(email="*****@*****.**", password="******")
        self.place = Place(city_id=self.city.id,
                           user_id=self.user.id,
                           name="",
                           description="")
        if (os.getenv('HBNB_TYPE_STORAGE') == 'db'):
            self.session.new(self.state)
            self.session.new(self.city)
            self.session.new(self.user)
            self.session.new(self.place)
            self.session.save()

コード例 #6

#!/usr/bin/python3
from os import getenv
from models.base_model import BaseModel
from models.user import User
from models.amenity import Amenity
from models.city import City
from models.place import Place
from models.state import State
from models.review import Review
from models.engine import file_storage
from models.engine import db_storage

if getenv('HBNB_TYPE_STORAGE') == "db":
    storage = db_storage.DBStorage()
else:
    storage = file_storage.FileStorage()
storage.reload()

コード例 #7

ファイル: test_db_storage.py プロジェクト: tope628/AirBnB_clone_v3

 def setUp(self):
     self.session = db_storage.DBStorage()
     self.session.reload()