def count(self, cls=None): ''' A method to count the number of objects in storage. ''' c = util.convert_class(cls) if c is None: return len(self.all()) else: return len(self.all(cls))
def get(self, cls, id): ''' A method to retrieve one object. ''' c = util.convert_class(cls) key = "{}.{}".format(c, id) obj = self.__objects.get(key) return obj
def get(self, cls, id): ''' A method to retrieve one object. Args: cls: string representing the class name. id: string representing the object ID. ''' c = util.convert_class(cls, "class") obj = self.__session.query(c).filter_by(id=id).first() return (obj)
def all(self, cls=None): ''' Return the dictionary ''' new_dict = {} cls = util.convert_class(cls) if cls is not None: for k, v in self.__objects.items(): if cls == k.split(".")[0]: new_dict[k] = v return new_dict else: return self.__objects
def all(self, cls=None): ''' Query current database session ''' db_dict = {} cls = util.convert_class(cls) if cls is not None: objs = self.__session.query(models.classes[cls]).all() for obj in objs: key = "{}.{}".format(obj.__class__.__name__, obj.id) db_dict[key] = obj return db_dict else: for k, v in models.classes.items(): if k != "BaseModel": objs = self.__session.query(v).all() if len(objs) > 0: for obj in objs: key = "{}.{}".format(obj.__class__.__name__, obj.id) db_dict[key] = obj return db_dict