def test_load_file_env(self): self.assertRaises(AssertionError, lambda: config.DB_CONNECTION) with mock.patch("os.getenv", return_value=self.fname): cfg.load_config() self.assertIsNotNone(config.DB_CONNECTION)
def login(website, username, password): """auto login and return cookie if success return Error code if fails: '1': failed to load config '2': timeout '3': cookie not found in headers or response """ cfg_data = load_config(website) if cfg_data is None: # error return '1' method = 'POST' headers = cfg_data['headers'] # body login_data = cfg_data.get('login_data', dict()) login_data[cfg_data['field_user']] = username login_data[cfg_data['field_password']] = password authcode_name = cfg_data['field_authcode'] if authcode_name: headers['Cookie'], login_data[authcode_name] = get_authcode(cfg_data['url_authcode']) body = urllib.urlencode(login_data) h = Http(timeout=T_TIMEOUT) try: rsp, content = h.request(cfg_data['url_login'], method, headers=headers, body=body) # response 302 except socket.timeout: return '2' # parse cookie clean_cookie_meth = getattr(clean_cookie, website, clean_cookie.raw) return clean_cookie_meth(rsp, content) or headers.get('Cookie', '3')
def configure_app(app): """Apply the configuration data from the config module to the flask application. :type app: flask.Flask :param app: The Application to configure. """ if not cfg.config.loaded(): try: cfg.load_config() except AssertionError: cfg.load_config(EXAMPLE_CONFIG) app.config.update(cfg.config.data) event_date = parse_cfg_date(app.config["EVENT_DATE"]) app.config["EVENT_DATE"] = event_date app.config["EVENT_DATE_PRETTY"] = pretty_date(event_date, month_name=True) app.config["EVENT_DATE_PRETTY_LONG"] = pretty_date(event_date, month_name=True, show_year=True, with_weekday=True) app.config["REGISTER_END"] = parse_cfg_date(app.config["REGISTER_END"])
def __call__(self, parser, namespace, values, option_string=None): values = resolve(values) scfg = cfg.load_config(values) if 'target' in scfg: if not hasattr(namespace, 'target'): setattr(namespace, 'target', wrt(values, scfg['target'])) if 'source' in scfg: if not hasattr(namespace, 'source'): setattr(namespace, 'source', wrt(values, scfg['source'])) if 'align' in scfg: if not hasattr(namespace, 'align'): setattr(namespace, 'align', wrt(values, scfg['align'])) if 'subcorpora' in scfg: setattr(namespace, 'subcorpora', scfg['subcorpora'])
def __call__(self,parser, namespace, values, option_string=None): values = resolve(values) scfg = cfg.load_config(values) if 'target' in scfg: if not hasattr(namespace,'target'): setattr(namespace,'target',wrt(values,scfg['target'])) if 'source' in scfg: if not hasattr(namespace,'source'): setattr(namespace,'source',wrt(values,scfg['source'])) if 'align' in scfg: if not hasattr(namespace,'align'): setattr(namespace,'align',wrt(values,scfg['align'])) if 'subcorpora' in scfg: setattr(namespace,'subcorpora',scfg['subcorpora'])
def main(args): print("Number of available GPUs: {}".format(torch.cuda.device_count())) device = torch.device('cpu') print("device", device) # config config = cfg.load_config(args.config) config.opt.n_iters_per_epoch = config.opt.n_objects_per_epoch // config.opt.batch_size model = { "alg": algebrictriangulation, "vol": volumetrictriangulation }[config.model.name](config, device=device).to(device) if config.model.name == "vol": criterion = L1loss else: criterion = MSEloss
import database as db from cfg import config from database.model import Team from geotools import simple_distance from geotools.routing import MapPoint arguments = argparse.ArgumentParser() arguments.add_argument("-c", "--config", help="set the configfile", default="config.yaml") arguments.add_argument("max_teams", help="Max. number of teams to export") args = arguments.parse_args() print "load config..." cfg.load_config(args.config) MAX_TEAMS = args.max_teams print "init db..." db.init_session(connection_string=config.DB_CONNECTION) print "fetch teams..." teams = db.session.query(Team).filter_by(deleted=False).filter_by(confirmed=True, backup=False).order_by(Team.id).limit( MAX_TEAMS).all() data = [] round_teams = defaultdict(list) max_working = len(teams) - (len(teams) % 3) divider = max_working / 3.0
help= "A file with a list of space, comma or semicolon separated recipients") spam.set_defaults(func=cmd_send_spam) return args.parse_args() def cmd_send_informal(args): informal_to_teams(args.templatename, args.subject, not args.nodebug) def cmd_send_planmails(args): results = read_plan_file(args) plans_to_teams(results, not args.nodebug, args.group, args.separate, args.exclude) def cmd_send_emergency_routes(args): results = read_plan_file(args) emergency_plan_routes(results, not args.nodebug) def cmd_send_spam(args): send_spam(args.addresses, not args.nodebug) if __name__ == "__main__": args = parse_args() cfg.load_config(args.config) args.func(args)
def main(): # load config config = load_config() _consumer_key = "" _consumer_secret = "" if config.has_section('client'): _consumer_key = config.get('client', '_consumer_key') _consumer_secret = config.get('client', '_consumer_secret') # init client client = Client(_consumer_key, _consumer_secret) # auth before use auth(client, config) if client.is_authed(): global local_path global g_ind if config.has_section('sync'): local_path = config.get('sync', 'local_path') if not local_path: home_dir = os.path.expanduser('~') default_dir = os.path.join(home_dir, "kuaipan") is_create_default = False if not os.path.exists(default_dir): is_create_default = True os.mkdir(default_dir) dialog = gtk.FileChooserDialog(title="请选择快盘文件同步到本地存放的文件夹", parent=None, action=gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER, buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN,gtk.RESPONSE_OK)) dialog.set_current_folder(default_dir) filter = gtk.FileFilter() filter.set_name("所有文件夹") filter.add_pattern("*") dialog.add_filter(filter) response = dialog.run() if response == gtk.RESPONSE_OK: print dialog.get_filename(), 'selected' local_path = dialog.get_filename() if is_create_default and local_path != default_dir: os.rmdir(default_dir) config.add_section('sync') config.set('sync', 'local_path', local_path) fp = open(get_cfg_path(), 'w') config.write(fp) else: print 'Closed, no files selected' dialog.destroy() if not local_path: return ac_info = client.get_account_info() print('is authed') print(ac_info) window = gtk.Window(gtk.WINDOW_TOPLEVEL) g_ind = init_indicator(ac_info, client) # start sync thread start_sync(client, local_path, g_ind) # start monitor thread start_monitor(client, local_path, g_ind) # start run application gtk.main() else: print('not authed')
def test_load_file_arg(self): self.assertRaises(AssertionError, lambda: config.DB_CONNECTION) self.assertRaises(AssertionError, cfg.load_config) cfg.load_config(self.fname) self.assertIsNotNone(config.DB_CONNECTION)
from fabric.api import * from fabric.utils import * from fabric.contrib.files import * from server_util import * import cfg site_map = cfg.load_config() env.user = '******' env.roledefs = site_map env.key_filename = 'aws_key.pem' @task def update_hosts(hostname, ip): host_util = hosts(get_lines("/etc/hosts")) text = host_util.update(ip, hostname) put(tmp(text), '/etc/hosts', use_sudo=True) sudo('chown root:root /etc/hosts') @task def config_hostname(hostname): sed('/etc/sysconfig/network', 'HOSTNAME=.*', 'HOSTNAME=%s' % hostname, use_sudo=True) sudo('hostname %s' % hostname) pass @task def base_install(): sudo('yum install -y telnet wget nfs-utils') sudo('chkconfig rpcbind on')
# # Update # @Ekira, 2017/7/27 # Fix "out of range" bug. # Add function: cycling showing images from a directory. # Add function: get/set wallpaper from bing's index image everyday. # TODO: modebingindex.py, unsupport loopping everyday now. import sys, os from cfg import gen_config, load_config from modesingle import single from modecycle import cycle from modebingindex import bingindex if __name__ == '__main__': CONFIG_PATH = "./whitodo.cfg" if not os.path.exists(CONFIG_PATH): gen_config(CONFIG_PATH) sys.exit() imgcfg = load_config(CONFIG_PATH) print imgcfg if imgcfg["mode"] == "mode-single": single(imgcfg) elif imgcfg["mode"] == "mode-cycle": cycle(imgcfg) elif imgcfg["mode"] == "mode-bingindex": bingindex(imgcfg) else: sys.exit()