本文整理汇总了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.
#.........这里部分代码省略.........