py-gitea/gitea/gitea.py

402 wiersze
17 KiB
Python

import json
import requests
import logging
class Organization:
# ORG_CREATE = """/admin/users/%s/orgs""" # <org>
# ORG_CREATE = """/orgs""" # <org>
ORG_REQUEST = """/orgs/%s""" # <org>
ORG_REPOS_REQUEST = """/orgs/%s/repos""" # <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):
self.gitea = gitea
self.username = "UNINIT"
self.__initialize_org(orgName, initJson)
def get_repositories(self):
results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST %
self.username)
return [Repository(self.gitea, self, result["name"],
initJson=result) for result in results]
def get_teams(self):
results = self.gitea.requests_get(Organization.ORG_TEAMS_REQUEST %
self.username)
return [Team(self, result["name"], initJson=result)
for result in results]
def get_members(self):
results = self.gitea.requests_get(Organization.ORG_GET_MEMBERS %
self.username)
return [User(self, result["username"], initJson=result)
for result in results]
def __initialize_org(self, orgName: str, result) -> None:
try:
if not result:
result = self.gitea.requests_get(Organization.ORG_REQUEST %
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):
result = self.gitea.requests_patch(Organization.ORG_PATCH %
self.username, data=values)
self.__initialize_org(self.username, result)
class User:
USER_REQUEST = """/users/%s""" # <org>
USER_REPOS_REQUEST = """/users/%s/repos""" # <org>
USER_PATCH = """/admin/users/%s""" # <username>
ADMIN_DELETE_USER = """/admin/users/%s""" # <username>
def __init__(self, gitea, userName: str, initJson: json = None):
self.gitea = gitea
self.username = "UNINIT"
self.__initialize_user(userName, initJson)
def get_repositories(self):
result = self.gitea.requests_get(User.USER_REPOS_REQUEST %
self.username)
return [Repository(self.gitea, self, r["name"]) for r in result]
def __initialize_user(self, userName: str, result) -> None:
if not result:
result = self.gitea.requests_get(User.USER_REQUEST % userName)
logging.info("Found User: %s" % userName)
for i, v in result.items():
setattr(self, i, v)
def set_value(self, email: str, values: dict):
# the request requires email to be set...
values["email"] = email
result = self.gitea.requests_patch(User.USER_PATCH % self.username,
data=values)
self.__initialize_user(self.username, result)
def delete(self):
self.gitea.requests_delete(User.ADMIN_DELETE_USER % self.username)
class Repository:
REPO_REQUEST = """/repos/%s/%s""" # <ownername>,<reponame>
REPO_SEARCH = """/repos/search/%s""" # <reponame>
REPO_BRANCHES = """/repos/%s/%s/branches""" # <owner>, <reponame>
def __init__(self, gitea, repoOwner, repoName: str, initJson: json = None):
self.gitea = gitea
self.name = "UNINIT"
self.__initialize_repo(repoOwner, repoName, initJson)
def __initialize_repo(self, repoOwner, repoName: str, result):
if not result:
result = self.gitea.requests_get(Repository.REPO_REQUEST %
(repoOwner.username, repoName))
logging.info("Found Repository: %s/%s" %
repoOwner.username, repoName)
for i, v in result.items():
setattr(self, i, v)
self.owner = repoOwner
def get_branches(self):
results = self.gitea.requests_get(Repository.REPO_BRANCHES %
self.owner.username, self.name)
return [Branch(self, result["name"], result) for result in results]
class Branch:
REPO_BRANCH = """/repos/%s/%s/branches/%s""" # <owner>, <repo>, <ref>
def __init__(self, repo: Repository, name: str, initJson: json = None):
self.gitea = repo.gitea
self.__initialize_branch(repo, name, initJson)
def __initialize_branch(self, repository, name, result):
if not result:
result = self.gitea.requests_get(Branch.REPO_BRANCH %
(repository.owner.username,
repository.name, name))
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
class Team:
def __init__(self, org: Organization, name: str, initJson: json = None):
self.gitea = org.gitea
self.__initialize_repo(org, name, initJson)
def __initialize_repo(self, org, name, result):
if not result:
raise NotImplementedError("TODO")
logging.info("Team found: %s/%s" % (org.username, name))
for i, v in result.items():
setattr(self, i, v)
self.organization = org
class Gitea():
ADMIN_CREATE_USER = """/admin/users"""
ADMIN_REPO_CREATE = """/admin/users/%s/repos""" # <ownername>
GITEA_VERSION = """/version"""
GET_USER = """/user"""
# CREATE_ORG = """/orgs"""
CREATE_ORG = """/admin/users/%s/orgs""" # username
"""
@:param url: url of Gitea server without .../api/<version>
"""
def __init__(self, url, token):
self.headers = {"Authorization": "token " + token}
self.url = url
self.requests = requests
def get_url(self, endpoint):
url = self.url + "/api/v1" + endpoint
logging.info(url)
return url
def parse_result(self, result):
if (result.text and len(result.text) > 3):
return json.loads(result.text)
return {}
def requests_get(self, endpoint):
request = self.requests.get(self.get_url(endpoint),
headers=self.headers)
if request.status_code not in [200, 201]:
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)
def requests_delete(self, endpoint):
request = self.requests.delete(self.get_url(endpoint),
headers=self.headers)
if request.status_code not in [204]:
logging.error("Received status code: %s (%s)" %
(request.status_code, request.url))
raise Exception("Received status code: %s (%s)" %
(request.status_code, request.url))
def requests_post(self, endpoint, data):
request = self.requests.post(self.get_url(endpoint),
headers=self.headers, data=data)
if request.status_code not in [200, 201]:
logging.error("Received status code: %s (%s)" %
(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)
def requests_patch(self, endpoint, data):
request = self.requests.patch(self.get_url(endpoint),
headers=self.headers, data=data)
if request.status_code not in [200, 201]:
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)
def get_users_search(self, ):
path = '/users/search'
return self.requests_get(path)
def delete_repos(self, username, reponame):
path = '/repos/' + username + '/' + reponame
return self.requests.delete(path)
def get_orgs_public_members_all(self, orgname):
path = '/orgs/' + orgname + '/public_members'
return self.requests_get(path)
def post_repos__forks(self, organization, repo, owner):
path = '/repos/' + owner + '/' + repo + '/forks'
return self.requests_post(path, data={'organization': organization})
def get_repos_forks(self, repo, owner):
path = '/repos/' + owner + '/' + repo + '/forks'
return self.requests_get(path)
def put_repos__subscription(self, username, reponame):
path = '/repos/' + username + '/' + reponame + '/subscription'
return self.requests.put(path)
def delete_repos_subscription(self, username, reponame):
path = '/repos/' + username + '/' + reponame + '/subscription'
return self.requests.delete(path)
def get_repos_subscription(self, username, reponame):
path = '/repos/' + username + '/' + reponame + '/subscription'
return self.requests_get(path)
def get_users_following(self, username):
path = '/users/' + username + '/following'
return self.requests_get(path)
def get_users_starred(self, username):
path = '/users/' + username + '/starred'
return self.requests_get(path)
def put_orgs_public_members(self, username, orgname):
path = '/orgs/' + orgname + '/public_members/' + username
return self.requests.put(path)
def delete_orgs_public_members(self, username, orgname):
path = '/orgs/' + orgname + '/public_members/' + username
return self.requests.delete(path)
def get_orgs_public_members(self, username, orgname):
path = '/orgs/' + orgname + '/public_members/' + username
return self.requests_get(path)
def post_org_repos(self, name, description, private, auto_init, gitignores,
license, readme, org):
path = '/org/' + org + '/repos'
return self.requests_post(path, data={'name': name,
'description': description,
'private': private,
'auto_init': auto_init,
'gitignores': gitignores,
'license': license,
'readme': readme})
def delete_orgs_members(self, orgname, username):
path = '/orgs/' + orgname + '/members/' + username
return self.requests.delete(path)
def post_repos__hooks(self, type, config, events, active,
reponame, username):
path = '/repos/' + username + '/' + reponame + '/hooks'
return self.requests_post(path, data={'type': type, 'config': config,
'events': events,
'active': active})
def get_repos_hooks(self, reponame, username):
path = '/repos/' + username + '/' + reponame + '/hooks'
return self.requests_get(path)
def post_repos_migrate(self, clone_addr, auth_username, auth_password, uid,
repo_name, mirror, private,
description):
path = '/repos/migrate'
return self.requests_post(path, data={'clone_addr': clone_addr,
'auth_username': auth_username,
'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):
path = '/user/repos'
return self.requests_post(path, data={'name': name,
'description': description,
'private': private,
'auto_init': auto_init,
'gitignores': gitignores,
'license': license,
'readme': readme})
# # #
def get_user(self) -> User:
result = self.requests_get(Gitea.GET_USER)
return User(self, "UNINIT", initJson=result)
def get_version(self) -> str:
result = self.requests_get(Gitea.GITEA_VERSION)
return result["version"]
def create_user(self, userName: str, email: str, fullName: str,
password: str, sendNotify=True, sourceId=0) \
-> User:
result = self.requests_post(Gitea.ADMIN_CREATE_USER,
data={'source_id': sourceId,
'login_name': userName,
'username': userName,
'full_name': fullName,
'email': email,
'password': password,
'send_notify': sendNotify})
if "id" in result:
logging.info("Successfully created User %s (id %s)" %
result["login"], result["id"])
else:
logging.error(result["message"])
raise Exception("User not created... (gitea: %s)" %
result["message"])
return User(self, userName, result)
def create_repo(self, repoOwner, repoName: str, description: str,
private: bool, autoInit=True, gitignores=None,
license=None, readme="Default") -> Repository:
# although this only says user in the api, this also works for
# organizations
assert(isinstance(repoOwner, User) or
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:
logging.info("Successfully created Repository %s " %
result["name"])
else:
logging.error(result["message"])
raise Exception("Repository not created... (gitea: %s)" %
result["message"])
return Repository(self, repoOwner, repoName, result)
def create_org(self, owner: User, orgName: str, description: str,
location="") -> Organization:
assert (isinstance(owner, User))
result = self.requests_post(Gitea.CREATE_ORG % owner.username,
# result = self.requests_post(Gitea.CREATE_ORG,
data={"username": orgName,
"description": description,
"location": location,
"website": "", "full_name": ""})
if "id" in result:
logging.info("Successfully created Repository %s" %
(result["name"]))
else:
logging.error(result["message"])
raise Exception("Repository not created... (gitea: %s)" %
(result["message"]))
return Organization(owner, orgName, initJson=result)