Mercurial > kallithea
view rhodecode/bin/base.py @ 4068:17e3926c3753
added ini files into codecleaner
author | Marcin Kuzminski <marcin@python-works.com> |
---|---|
date | Sat, 29 Jun 2013 23:00:10 +0200 |
parents | 5a7d52cf084d |
children | ffd45b185016 |
line wrap: on
line source
""" Base utils for shell scripts """ import os import sys import random import urllib2 import pprint try: from rhodecode.lib.ext_json import json except ImportError: try: import simplejson as json except ImportError: import json CONFIG_NAME = '.rhodecode' 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 !') 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