def get(select_filter):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
return db_controller.as_array("users", select_filter)
def __save_photo(self):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
handle = HandlePhotoUpload(db_controller.get_database())
return handle.put(self.get_photo())
def save(self):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
insert_result = db_controller.insert("sources", {
"url": self.get_url(),
"user": self.get_user()
})
return insert_result.inserted_id is not None
def __get_photo_stream(self):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
handle = HandlePhotoUpload(db_controller.get_database())
file_id = self.get({
"username": self.get_username(),
"password": self.get_password()
})[0]["photo"]
return handle.write_on_file(file_id)
def save(self):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
insert_result = db_controller.insert(
"users", {
"name": self.__name,
"username": self.__username,
"password": self.__password,
"email": self.__email,
"photo": self.__save_photo()
})
return insert_result.inserted_id is not None
def save(self):
if os.environ.get("MONGODB_URI") is None:
db_controller = DBController()
else:
db_controller = DBController(uri=os.environ.get("MONGODB_URI"))
db_controller.set_dbname(os.environ.get("DATABASE"))
db_controller.connect()
self.set_date(datetime.now().timestamp())
insert_result = db_controller.insert("articles",
{
"name": self.__name,
"source": self.__src,
"content": self.__content,
"user": self.__user,
"date": self.get_date()
})
return insert_result.inserted_id is not None
