From 37123e9ca25778d53b675626ddc32ff2e3e04314 Mon Sep 17 00:00:00 2001 From: Guido Günther Date: Tue, 15 Jul 2014 21:57:12 +0100 Subject: Initial commit --- industriart/__init__.py | 4 ++ industriart/artifactory.py | 142 +++++++++++++++++++++++++++++++++++++++++++++ industriart/compat.py | 13 +++++ 3 files changed, 159 insertions(+) create mode 100644 industriart/__init__.py create mode 100644 industriart/artifactory.py create mode 100644 industriart/compat.py (limited to 'industriart') diff --git a/industriart/__init__.py b/industriart/__init__.py new file mode 100644 index 0000000..e058189 --- /dev/null +++ b/industriart/__init__.py @@ -0,0 +1,4 @@ +import logging +from .compat import NullHandler + +logging.getLogger(__name__).addHandler(NullHandler()) diff --git a/industriart/artifactory.py b/industriart/artifactory.py new file mode 100644 index 0000000..241d7ba --- /dev/null +++ b/industriart/artifactory.py @@ -0,0 +1,142 @@ +# vim: set fileencoding=utf-8 : +# +# (C) 2014 Guido Gunther +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +import logging +import os +import posixpath +import requests + +log = logging.getLogger(__name__) + +class ArtifactoryError(Exception): + def __init__(self, msg, url, status_code=0): + self.url = url + self.status_code = status_code + Exception.__init__(self, msg) + +class ArtifactoryObjectNotFound(ArtifactoryError): + """Object not found aka 404""" + pass + +class ArtifactoryNoPermission(ArtifactoryError): + """Object not found aka 401""" + pass + +class Artifactory(object): + """ + Extremely simple JFrog Artifactory API Wrapper + """ + def __init__(self, base, user=None, password=None): + self.user = user + self.password = password + self.base = base + + def search_gavc(self, groupid=None, artifactid=None, version=None, classifier=None): + """ + Query the artifactory via Groupid, ArtifacId, Version + """ + path='api/search/gavc' + param_map = { + 'groupid': 'g', + 'artifactid': 'a', + 'version': 'v', + 'classifier': 'c', + } + params = {} + for k,v in locals().items(): + if k in param_map and v is not None: + params[param_map[k]] = v + + url = posixpath.join(self.base, path) + log.debug("Searching on %s with %s" % (url, params)) + return self.get(url, params)['results'] + + + def calc_yum_metadata(self, repository): + url = posixpath.join(self.base, 'api/yum', repository) + params = { 'async': 0 } + log.debug("Refreshing on %s" % url) + return self.post(url, params) + + def get_repositories(self): + url = posixpath.join(self.base, 'api/repositories') + return self.get(url) + + def _transform_url(self, url): + """ + Helper to rewrite URLs. All artifactory URLs are passed through this method + in order to cope with broken installations where e.g. artifactory sits behind + a reverse proxy but doesn't know about it. + + Rewrite this in derived classes as needed (but hopefully not). + """ + return url + + + @staticmethod + def _parse_error(data): + try: + if data.has_key('errors'): + return data['errors'][0]['message'] + return None + except: + return None + + + def _request(self, http_method, url, params=None): + """ + Perform a HTTP request on the artifactory + """ + url = self._transform_url(url) + auth = None + + if self.user and self.password: + auth = requests.auth.HTTPBasicAuth(self.user, self.password) + + method = getattr(requests, http_method.lower()) + r = method(url, params=params, auth=auth) + if r.status_code / 100 != 2: + err = self._parse_error(r.json()) + if r.status_code == 404: + raise ArtifactoryObjectNotFound(err or "Not found" % url, + url, + r.status_code) + elif r.status_code == 401: + raise ArtifactoryNoPermission(err or "Unauthorized" % url, + url, + r.status_code) + raise ArtifactoryError(err or "Did not get a 2xx", + url, + r.status_code) + if r.headers['content-type'].endswith('json'): + return r.json() + else: + return r.text + + + def get(self, url, params=None): + """ + Perform a GET request on the artifactory + """ + return self._request('GET', url, params) + + + def post(self, url, params): + """ + Perform a POST request on the artifactory + """ + return self._request('POST', url, params) diff --git a/industriart/compat.py b/industriart/compat.py new file mode 100644 index 0000000..06fe440 --- /dev/null +++ b/industriart/compat.py @@ -0,0 +1,13 @@ +"""Compatiblity with older python""" + +import sys + +# python 2.7 introduced a NullHandler which we want to use, but to support +# older versions, we implement our own if needed. +if sys.version_info[:2] > (2, 6): + from logging import NullHandler +else: + from logging import Handler + class NullHandler(Handler): + def emit(self, record): + pass -- cgit v1.2.3