本文整理匯總了Python中core.db.connection.DataHubConnection.close_connection方法的典型用法代碼示例。如果您正苦於以下問題：Python DataHubConnection.close_connection方法的具體用法？Python DataHubConnection.close_connection怎麽用？Python DataHubConnection.close_connection使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類core.db.connection.DataHubConnection的用法示例。
在下文中一共展示了DataHubConnection.close_connection方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: _superuser_connection
# 需要導入模塊: from core.db.connection import DataHubConnection [as 別名]
# 或者: from core.db.connection.DataHubConnection import close_connection [as 別名]
class _superuser_connection():
superuser_con = None
def __init__(self, repo_base=None):
self.repo_base = repo_base
def __enter__(self):
self.superuser_con = DataHubConnection(
user=settings.DATABASES['default']['USER'],
password=settings.DATABASES['default']['PASSWORD'],
repo_base=self.repo_base)
return self.superuser_con
def __exit__(self, type, value, traceback):
self.superuser_con.close_connection()示例2: __init__
# 需要導入模塊: from core.db.connection import DataHubConnection [as 別名]
# 或者: from core.db.connection.DataHubConnection import close_connection [as 別名]
class DataHubManager:
def __init__(self, user=settings.ANONYMOUS_ROLE, repo_base=None,
is_app=False):
# blank users are set to anonymous role
if user == '':
user = settings.ANONYMOUS_ROLE
username = None
password = None
if is_app:
app = App.objects.get(app_id=user)
username = app.app_id
password = hashlib.sha1(app.app_token).hexdigest()
else:
user = User.objects.get(username=user)
username = user.username
password = user.password
if not repo_base:
repo_base = username
self.username = username
self.repo_base = repo_base
self.user_con = DataHubConnection(
user=username,
repo_base=repo_base,
password=password)
def __enter__(self):
return self
def __exit__(self, type, value, traceback):
self.close_connection()
""" Basic Operations. """
def change_repo_base(self, repo_base):
"""Changes the repo base and resets the DB connection."""
self.user_con.change_repo_base(repo_base=repo_base)
def close_connection(self):
self.user_con.close_connection()
def create_repo(self, repo):
"""
Creates a repo in the current repo_base.
Returns True on success.
Succeeds if repo already exists.
Raises ValueError on an invalid repo name.
Raises ProgrammingError on permission denied.
"""
return self.user_con.create_repo(repo=repo)
def list_repos(self):
"""
Returns a list of repo (schema) names in the current repo_base.
Should never fail or raise any exceptions.
"""
return sorted(self.user_con.list_repos())
def rename_repo(self, repo, new_name):
"""
Renames a repo.
Returns True on success.
Raises ValueError if either name has invalid characters.
Raises LookupError if the repo doesn't exist.
Raises ValueError if the new name is taken.
Raises PermissionDenied if not repo_base owner.
"""
# only a repo owner can rename a repo:
if self.repo_base != self.username:
raise PermissionDenied()
# rename in user_con
success = self.user_con.rename_repo(repo=repo, new_name=new_name)
if success:
# update collaborator(s), if there are any
Collaborator.objects.filter(
repo_name=repo, repo_base=self.repo_base).update(
repo_name=new_name)
return success
def list_collaborator_repos(self):
"""
Lists repos to which the current user has been granted access.
Returns Collaborator objects, which have repo_base and repo attributes.
Includes repos across all databases, not just the DataHubManager's
repo_base.
#.........這裏部分代碼省略.........