Mercurial > kallithea
view kallithea/bin/base.py @ 4208:ad38f9f93b3b kallithea-2.2.5-rebrand
Correct licensing information in individual files.
The top-level license file is now LICENSE.md.
Also, in various places where there should have been joint copyright holders
listed, a single copyright holder was listed. It does not appear easy to add
a link to a large list of copyright holders in these places, so it simply
refers to the fact that various authors hold copyright.
In future, if an easy method is discovered to link to a list from those
places, we should do so.
Finally, text is added to LICENSE.md to point to where the full list of
copyright holders is, and that Kallithea as a whole is GPLv3'd.
author | Bradley M. Kuhn <bkuhn@sfconservancy.org> |
---|---|
date | Wed, 21 May 2014 16:59:37 -0400 |
parents | d1addaf7a91e |
children | 1948ede028ef |
line wrap: on
line source
# -*- coding: utf-8 -*- # 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 3 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, see <http://www.gnu.org/licenses/>. """ kallithea.bin.base ~~~~~~~~~~~~~~~~~~ Base utils for shell scripts :created_on: May 09, 2013 :author: marcink :copyright: (c) 2013 RhodeCode GmbH. :license: GPLv3, see LICENSE.md for more details. """ import os import sys import random import urllib2 import pprint try: from kallithea.lib.ext_json import json except ImportError: try: import simplejson as json except ImportError: import json CONFIG_NAME = '.config/kallithea' FORMAT_PRETTY = 'pretty' FORMAT_JSON = 'json' def api_call(apikey, apihost, method=None, **kw): """ Api_call wrapper for RhodeCode. :param apikey: :param apihost: :param format: formatting, pretty means prints and pprint of json json returns unparsed json :param method: :returns: json response from server """ def _build_data(random_id): """ Builds API data with given random ID :param random_id: """ return { "id": random_id, "api_key": apikey, "method": method, "args": kw } if not method: raise Exception('please specify method name !') apihost = apihost.rstrip('/') id_ = random.randrange(1, 9999) req = urllib2.Request('%s/_admin/api' % apihost, data=json.dumps(_build_data(id_)), headers={'content-type': 'text/plain'}) ret = urllib2.urlopen(req) raw_json = ret.read() json_data = json.loads(raw_json) id_ret = json_data['id'] if id_ret == id_: return json_data else: _formatted_json = pprint.pformat(json_data) raise Exception('something went wrong. ' 'ID mismatch got %s, expected %s | %s' % ( id_ret, id_, _formatted_json)) class RcConf(object): """ RhodeCode config for API conf = RcConf() conf['key'] """ def __init__(self, config_location=None, autoload=True, autocreate=False, config=None): HOME = os.getenv('HOME', os.getenv('USERPROFILE')) or '' HOME_CONF = os.path.abspath(os.path.join(HOME, CONFIG_NAME)) self._conf_name = HOME_CONF if not config_location else config_location self._conf = {} if autocreate: self.make_config(config) if autoload: self._conf = self.load_config() def __getitem__(self, key): return self._conf[key] def __nonzero__(self): if self._conf: return True return False def __eq__(self): return self._conf.__eq__() def __repr__(self): return 'RcConf<%s>' % self._conf.__repr__() def make_config(self, config): """ Saves given config as a JSON dump in the _conf_name location :param config: """ update = False if os.path.exists(self._conf_name): update = True with open(self._conf_name, 'wb') as f: json.dump(config, f, indent=4) if update: sys.stdout.write('Updated config in %s\n' % self._conf_name) else: sys.stdout.write('Created new config in %s\n' % self._conf_name) def update_config(self, new_config): """ Reads the JSON config updates it's values with new_config and saves it back as JSON dump :param new_config: """ config = {} try: with open(self._conf_name, 'rb') as conf: config = json.load(conf) except IOError, e: sys.stderr.write(str(e) + '\n') config.update(new_config) self.make_config(config) def load_config(self): """ Loads config from file and returns loaded JSON object """ try: with open(self._conf_name, 'rb') as conf: return json.load(conf) except IOError, e: #sys.stderr.write(str(e) + '\n') pass