def adjust_database_uri(cls, uri: URL, selected_schema: Optional[str] = None) -> URL: if selected_schema: uri = uri.set(database=parse.quote(selected_schema, safe="")) return uri
def adjust_database_uri(cls, uri: URL, selected_schema: Optional[str] = None) -> URL: database = uri.database if "/" in uri.database: database = uri.database.split("/")[0] if selected_schema: selected_schema = parse.quote(selected_schema, safe="") uri = uri.set(database=f"{database}/{selected_schema}") return uri
def adjust_database_uri(cls, uri: URL, selected_schema: Optional[str] = None) -> URL: database = uri.database if selected_schema and database: selected_schema = parse.quote(selected_schema, safe="") if "/" in database: database = database.split("/")[0] + "/" + selected_schema else: database += "/" + selected_schema uri = uri.set(database=database) return uri