pull/1/head
fkarg 2019-03-17 12:19:18 +01:00
rodzic b1fd11cbd8
commit 13ee37bada
1 zmienionych plików z 179 dodań i 87 usunięć

Wyświetl plik

@ -5,76 +5,101 @@ import logging
class Organization: class Organization:
ORG_REQUEST = """/orgs/%s""" #<org> # ORG_CREATE = """/admin/users/%s/orgs""" # <org>
ORG_REPOS_REQUEST = """/orgs/%s/repos""" #<org> # ORG_CREATE = """/orgs""" # <org>
ORG_TEAMS_REQUEST = """/orgs/%s/teams""" #<org> ORG_REQUEST = """/orgs/%s""" # <org>
ORG_PATCH = """/orgs/%s""" #<org> ORG_REPOS_REQUEST = """/orgs/%s/repos""" # <org>
ORG_GET_MEMBERS = """/orgs/%s/members""" # <org> ORG_TEAMS_REQUEST = """/orgs/%s/teams""" # <org>
ORG_PATCH = """/orgs/%s""" # <org>
ORG_GET_MEMBERS = """/orgs/%s/members""" # <org>
def __init__(self, gitea, orgName : str, initJson: json = None): def __init__(self, gitea, orgName: str, initJson: json = None):
self.gitea = gitea self.gitea = gitea
self.username = "UNINIT" self.username = "UNINIT"
self.__initialize_org(orgName, initJson) self.__initialize_org(orgName, initJson)
def get_repositories(self): def get_repositories(self):
results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST%self.username) results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST %
return [Repository(self.gitea, self, result["name"], initJson=result) for result in results] self.username)
return [Repository(self.gitea, self, result["name"],
initJson=result) for result in results]
def get_teams(self): def get_teams(self):
results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST % self.username) results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST %
return [Team(self, result["name"], initJson=result) for result in results] self.username)
return [Team(self, result["name"], initJson=result)
for result in results]
def get_members(self): def get_members(self):
results = self.gitea.requests_get(Organization.ORG_GET_MEMBERS % self.username) results = self.gitea.requests_get(Organization.ORG_GET_MEMBERS %
return [User(self, result["username"], initJson=result) for result in results] self.username)
return [User(self, result["username"], initJson=result)
for result in results]
def __initialize_org(self, orgName: str, result) -> None: def __initialize_org(self, orgName: str, result) -> None:
if not result: try:
result = self.gitea.requests_get(Organization.ORG_REQUEST%orgName) if not result:
logging.info("Found Orgranization: %s"%orgName) result = self.gitea.requests_get(Organization.ORG_REQUEST %
for i,v in result.items(): setattr(self, i, v) orgName)
logging.info("Found Orgranization: %s" % orgName)
for i, v in result.items():
setattr(self, i, v)
except Exception:
logging.error("Did not find organisation: %s" % orgName)
# @staticmethod
# def create(gitea: Gitea, orgname: str, description: str = '', full_name:
# str = '', location: str = '', website: str = ''):
# gitea.requests_post(ORG_CREATE, data={'description': description,
# 'full_name': full_name, 'location': location, 'username': orgname,
# 'website': website})
def set_value(self, values: dict): def set_value(self, values: dict):
result = self.gitea.requests_patch(Organization.ORG_PATCH%self.username, data=values) result = self.gitea.requests_patch(Organization.ORG_PATCH %
self.username, data=values)
self.__initialize_org(self.username, result) self.__initialize_org(self.username, result)
class User: class User:
USER_REQUEST = """/users/%s""" #<org> USER_REQUEST = """/users/%s""" # <org>
USER_REPOS_REQUEST = """/users/%s/repos""" #<org> USER_REPOS_REQUEST = """/users/%s/repos""" # <org>
USER_PATCH = """/admin/users/%s""" # <username> USER_PATCH = """/admin/users/%s""" # <username>
ADMIN_DELETE_USER = """/admin/users/%s""" # <username> ADMIN_DELETE_USER = """/admin/users/%s""" # <username>
def __init__(self, gitea, userName : str, initJson: json = None): def __init__(self, gitea, userName: str, initJson: json = None):
self.gitea = gitea self.gitea = gitea
self.username = "UNINIT" self.username = "UNINIT"
self.__initialize_user(userName, initJson) self.__initialize_user(userName, initJson)
def get_repositories(self): def get_repositories(self):
result = self.gitea.requests_get(User.USER_REPOS_REQUEST%self.username) result = self.gitea.requests_get(User.USER_REPOS_REQUEST %
self.username)
return [Repository(self.gitea, self, r["name"]) for r in result] return [Repository(self.gitea, self, r["name"]) for r in result]
def __initialize_user(self, userName: str, result) -> None: def __initialize_user(self, userName: str, result) -> None:
if not result: if not result:
result = self.gitea.requests_get(User.USER_REQUEST % userName) result = self.gitea.requests_get(User.USER_REQUEST % userName)
logging.info("Found User: %s"%userName) logging.info("Found User: %s" % userName)
for i, v in result.items(): setattr(self, i, v) for i, v in result.items():
setattr(self, i, v)
def set_value(self, email: str, values: dict): def set_value(self, email: str, values: dict):
#the request requires email to be set... # the request requires email to be set...
values["email"] = email values["email"] = email
result = self.gitea.requests_patch(User.USER_PATCH%self.username, data=values) result = self.gitea.requests_patch(User.USER_PATCH % self.username,
data=values)
self.__initialize_user(self.username, result) self.__initialize_user(self.username, result)
def delete(self): def delete(self):
self.gitea.requests_delete(User.ADMIN_DELETE_USER%self.username) self.gitea.requests_delete(User.ADMIN_DELETE_USER % self.username)
class Repository: class Repository:
REPO_REQUEST = """/repos/%s/%s""" #<ownername>,<reponame> REPO_REQUEST = """/repos/%s/%s""" # <ownername>,<reponame>
REPO_SEARCH ="""/repos/search/%s""" # <reponame> REPO_SEARCH = """/repos/search/%s""" # <reponame>
REPO_BRANCHES = """/repos/%s/%s/branches""" #<owner>, <reponame> REPO_BRANCHES = """/repos/%s/%s/branches""" # <owner>, <reponame>
def __init__(self, gitea, repoOwner, repoName: str, initJson: json = None): def __init__(self, gitea, repoOwner, repoName: str, initJson: json = None):
self.gitea = gitea self.gitea = gitea
@ -83,19 +108,23 @@ class Repository:
def __initialize_repo(self, repoOwner, repoName: str, result): def __initialize_repo(self, repoOwner, repoName: str, result):
if not result: if not result:
result = self.gitea.requests_get(Repository.REPO_REQUEST % (repoOwner.username,repoName)) result = self.gitea.requests_get(Repository.REPO_REQUEST %
logging.info("Found Repository: %s/%s" % (repoOwner.username, repoName)) (repoOwner.username, repoName))
for i, v in result.items(): setattr(self, i, v) logging.info("Found Repository: %s/%s" %
repoOwner.username, repoName)
for i, v in result.items():
setattr(self, i, v)
self.owner = repoOwner self.owner = repoOwner
def get_branches(self): def get_branches(self):
results = self.gitea.requests_get(Repository.REPO_BRANCHES%(self.owner.username, self.name)) results = self.gitea.requests_get(Repository.REPO_BRANCHES %
self.owner.username, self.name)
return [Branch(self, result["name"], result) for result in results] return [Branch(self, result["name"], result) for result in results]
class Branch: class Branch:
REPO_BRANCH = """/repos/%s/%s/branches/%s""" # <owner>, <repo>, <ref> REPO_BRANCH = """/repos/%s/%s/branches/%s""" # <owner>, <repo>, <ref>
def __init__(self, repo: Repository, name: str, initJson: json = None): def __init__(self, repo: Repository, name: str, initJson: json = None):
self.gitea = repo.gitea self.gitea = repo.gitea
@ -104,11 +133,15 @@ class Branch:
def __initialize_branch(self, repository, name, result): def __initialize_branch(self, repository, name, result):
if not result: if not result:
result = self.gitea.requests_get(Branch.REPO_BRANCH % result = self.gitea.requests_get(Branch.REPO_BRANCH %
(repository.owner.username, repository.name, name)) (repository.owner.username,
logging.info("Branch found: %s/%s/%s" % (repository.owner.username, repository.name, name)) repository.name, name))
for i, v in result.items(): setattr(self, i, v) logging.info("Branch found: %s/%s/%s" % (repository.owner.username,
repository.name, name))
for i, v in result.items():
setattr(self, i, v)
self.repository = repository self.repository = repository
class Team: class Team:
def __init__(self, org: Organization, name: str, initJson: json = None): def __init__(self, org: Organization, name: str, initJson: json = None):
@ -119,17 +152,19 @@ class Team:
if not result: if not result:
raise NotImplementedError("TODO") raise NotImplementedError("TODO")
logging.info("Team found: %s/%s" % (org.username, name)) logging.info("Team found: %s/%s" % (org.username, name))
for i, v in result.items(): setattr(self, i, v) for i, v in result.items():
setattr(self, i, v)
self.organization = org self.organization = org
class Gitea(): class Gitea():
ADMIN_CREATE_USER = """/admin/users""" ADMIN_CREATE_USER = """/admin/users"""
ADMIN_REPO_CREATE = """/admin/users/%s/repos""" # <ownername> ADMIN_REPO_CREATE = """/admin/users/%s/repos""" # <ownername>
GITEA_VERSION = """/version""" GITEA_VERSION = """/version"""
GET_USER = """/user""" GET_USER = """/user"""
CREATE_ORG = """/admin/users/%s/orgs""" #username # CREATE_ORG = """/orgs"""
CREATE_ORG = """/admin/users/%s/orgs""" # username
""" """
@:param url: url of Gitea server without .../api/<version> @:param url: url of Gitea server without .../api/<version>
@ -150,30 +185,44 @@ class Gitea():
return {} return {}
def requests_get(self, endpoint): def requests_get(self, endpoint):
request = self.requests.get(self.get_url(endpoint), headers=self.headers) request = self.requests.get(self.get_url(endpoint),
if request.status_code not in [200,201]: headers=self.headers)
logging.error("Received status code: %s (%s)"%(request.status_code, request.url)) if request.status_code not in [200, 201]:
raise Exception(("Received status code: %s (%s)"%(request.status_code, request.url))) logging.error("Received status code: %s (%s)" %
(request.status_code, request.url))
raise Exception("Received status code: %s (%s)" %
(request.status_code, request.url))
return self.parse_result(request) return self.parse_result(request)
def requests_delete(self, endpoint): def requests_delete(self, endpoint):
request = self.requests.delete(self.get_url(endpoint), headers=self.headers) request = self.requests.delete(self.get_url(endpoint),
headers=self.headers)
if request.status_code not in [204]: if request.status_code not in [204]:
logging.error("Received status code: %s (%s)"%(request.status_code, request.url)) logging.error("Received status code: %s (%s)" %
raise Exception(("Received status code: %s (%s)"%(request.status_code, request.url))) (request.status_code, request.url))
raise Exception("Received status code: %s (%s)" %
(request.status_code, request.url))
def requests_post(self, endpoint, data): def requests_post(self, endpoint, data):
request = self.requests.post(self.get_url(endpoint), headers=self.headers, data=data) request = self.requests.post(self.get_url(endpoint),
headers=self.headers, data=data)
if request.status_code not in [200, 201]: if request.status_code not in [200, 201]:
logging.error("Received status code: %s (%s)"%(request.status_code, request.url)) logging.error("Received status code: %s (%s)" %
raise Exception(("Received status code: %s (%s)"%(request.status_code, request.url))) (request.status_code, request.url))
logging.error("With info: %s (%s)" %
(data, self.headers))
raise Exception("Received status code: %s (%s)" %
(request.status_code, request.url))
return self.parse_result(request) return self.parse_result(request)
def requests_patch(self, endpoint, data): def requests_patch(self, endpoint, data):
request = self.requests.patch(self.get_url(endpoint), headers=self.headers, data=data) request = self.requests.patch(self.get_url(endpoint),
headers=self.headers, data=data)
if request.status_code not in [200, 201]: if request.status_code not in [200, 201]:
logging.error("Received status code: %s (%s)"%(request.status_code, request.url)) logging.error("Received status code: %s (%s)" %
raise Exception(("Received status code: %s (%s)"%(request.status_code, request.url))) (request.status_code, request.url))
raise Exception("Received status code: %s (%s)" %
(request.status_code, request.url))
return self.parse_result(request) return self.parse_result(request)
def get_users_search(self, ): def get_users_search(self, ):
@ -228,35 +277,54 @@ class Gitea():
path = '/orgs/' + orgname + '/public_members/' + username path = '/orgs/' + orgname + '/public_members/' + username
return self.requests_get(path) return self.requests_get(path)
def post_org_repos(self, name, description, private, auto_init, gitignores, license, readme, org): def post_org_repos(self, name, description, private, auto_init, gitignores,
license, readme, org):
path = '/org/' + org + '/repos' path = '/org/' + org + '/repos'
return self.requests_post(path, data={'name': name, 'description': description, 'private': private, return self.requests_post(path, data={'name': name,
'auto_init': auto_init, 'gitignores': gitignores, 'license': license, 'description': description,
'private': private,
'auto_init': auto_init,
'gitignores': gitignores,
'license': license,
'readme': readme}) 'readme': readme})
def delete_orgs_members(self, orgname, username): def delete_orgs_members(self, orgname, username):
path = '/orgs/' + orgname + '/members/' + username path = '/orgs/' + orgname + '/members/' + username
return self.requests.delete(path) return self.requests.delete(path)
def post_repos__hooks(self, type, config, events, active, reponame, username): def post_repos__hooks(self, type, config, events, active,
reponame, username):
path = '/repos/' + username + '/' + reponame + '/hooks' path = '/repos/' + username + '/' + reponame + '/hooks'
return self.requests_post(path, data={'type': type, 'config': config, 'events': events, 'active': active}) return self.requests_post(path, data={'type': type, 'config': config,
'events': events,
'active': active})
def get_repos_hooks(self, reponame, username): def get_repos_hooks(self, reponame, username):
path = '/repos/' + username + '/' + reponame + '/hooks' path = '/repos/' + username + '/' + reponame + '/hooks'
return self.requests_get(path) return self.requests_get(path)
def post_repos_migrate(self, clone_addr, auth_username, auth_password, uid, repo_name, mirror, private, def post_repos_migrate(self, clone_addr, auth_username, auth_password, uid,
repo_name, mirror, private,
description): description):
path = '/repos/migrate' path = '/repos/migrate'
return self.requests_post(path, data={'clone_addr': clone_addr, 'auth_username': auth_username, return self.requests_post(path, data={'clone_addr': clone_addr,
'auth_password': auth_password, 'uid': uid, 'repo_name': repo_name, 'auth_username': auth_username,
'mirror': mirror, 'private': private, 'description': description}) 'auth_password': auth_password,
'uid': uid,
'repo_name': repo_name,
'mirror': mirror,
'private': private,
'description': description})
def post_user_repos(self, name, description, private, auto_init, gitignores, license, readme): def post_user_repos(self, name, description, private, auto_init,
gitignores, license, readme):
path = '/user/repos' path = '/user/repos'
return self.requests_post(path, data={'name': name, 'description': description, 'private': private, return self.requests_post(path, data={'name': name,
'auto_init': auto_init, 'gitignores': gitignores, 'license': license, 'description': description,
'private': private,
'auto_init': auto_init,
'gitignores': gitignores,
'license': license,
'readme': readme}) 'readme': readme})
# # # # # #
@ -269,41 +337,65 @@ class Gitea():
result = self.requests_get(Gitea.GITEA_VERSION) result = self.requests_get(Gitea.GITEA_VERSION)
return result["version"] return result["version"]
def create_user(self, userName: str, email: str, fullName: str, password: str, sendNotify = True, sourceId = 0)\ def create_user(self, userName: str, email: str, fullName: str,
password: str, sendNotify=True, sourceId=0) \
-> User: -> User:
result = self.requests_post(Gitea.ADMIN_CREATE_USER, result = self.requests_post(Gitea.ADMIN_CREATE_USER,
data={'source_id': sourceId, 'login_name': userName, 'username': userName, 'full_name': fullName, data={'source_id': sourceId,
'email': email, 'password': password, 'send_notify': sendNotify}) 'login_name': userName,
'username': userName,
'full_name': fullName,
'email': email,
'password': password,
'send_notify': sendNotify})
if "id" in result: if "id" in result:
logging.info("Successfully created User %s (id %s)"%(result["login"], result["id"])) logging.info("Successfully created User %s (id %s)" %
result["login"], result["id"])
else: else:
logging.error(result["message"]) logging.error(result["message"])
raise Exception("User not created... (gitea: %s)"%result["message"]) raise Exception("User not created... (gitea: %s)" %
result["message"])
return User(self, userName, result) return User(self, userName, result)
def create_repo(self, repoOwner, repoName: str, description: str, def create_repo(self, repoOwner, repoName: str, description: str,
private: bool, autoInit = True, gitignores = None, license= None, readme = "Default") -> Repository: private: bool, autoInit=True, gitignores=None,
# although this only says user in the api, this also works for organizations license=None, readme="Default") -> Repository:
assert(isinstance(repoOwner, User) or isinstance(repoOwner, Organization)) # although this only says user in the api, this also works for
result = self.requests_post(Gitea.ADMIN_REPO_CREATE%repoOwner.username, # organizations
data={'name': repoName, 'description': description, 'private': private, assert(isinstance(repoOwner, User) or
'auto_init': autoInit, 'gitignores': gitignores, 'license': license, 'readme': readme}) isinstance(repoOwner, Organization))
result = self.requests_post(Gitea.ADMIN_REPO_CREATE %
repoOwner.username,
data={'name': repoName,
'description': description,
'private': private,
'auto_init': autoInit,
'gitignores': gitignores,
'license': license,
'readme': readme})
if "id" in result: if "id" in result:
logging.info("Successfully created Repository %s "%(result["name"])) logging.info("Successfully created Repository %s " %
result["name"])
else: else:
logging.error(result["message"]) logging.error(result["message"])
raise Exception("Repository not created... (gitea: %s)"%(result["message"])) raise Exception("Repository not created... (gitea: %s)" %
result["message"])
return Repository(self, repoOwner, repoName, result) return Repository(self, repoOwner, repoName, result)
def create_org(self, owner: User, orgName: str, description: str, location = "") -> Organization: def create_org(self, owner: User, orgName: str, description: str,
location="") -> Organization:
assert (isinstance(owner, User)) assert (isinstance(owner, User))
result = self.requests_post(Gitea.CREATE_ORG%owner.username, result = self.requests_post(Gitea.CREATE_ORG % owner.username,
data={'username': orgName, "description" : description, "location" : location}) # result = self.requests_post(Gitea.CREATE_ORG,
data={"username": orgName,
"description": description,
"location": location,
"website": "", "full_name": ""})
if "id" in result: if "id" in result:
logging.info("Successfully created Repository %s "%(result["name"])) logging.info("Successfully created Repository %s" %
(result["name"]))
else: else:
logging.error(result["message"]) logging.error(result["message"])
raise Exception("Repository not created... (gitea: %s)"%(result["message"])) raise Exception("Repository not created... (gitea: %s)" %
(result["message"]))
return Organization(owner, orgName, initJson=result) return Organization(owner, orgName, initJson=result)