コード例 #1
0
ファイル: generators.py プロジェクト: HyFrmn/Wrangler
def RenderJob(name='No Name',
              command = 'printf "Frame: %4s %s\n" $WRANGLER_FRAME $WRANGLER_PADFRAME',
              start=1,
              end=100,
              step=1,
              padding=4,
              priority=500,
              env=None,
              owner=None,
              **meta):


    config = config_base()
    if not env:
        env = {}
    env = update_env(env)
    if not owner:
        owner = os.environ['USER']
    job = Job(name, env=env)
    job.owner = owner
    for i in range(start, end + 1, step):
        task = Task(command)
        task.env.update(env)
        task.env['WRANGLER_PADFRAME'] = str(i).zfill(padding)
        task.env['WRANGLER_FRAME'] = i
        task.priority = priority
        job.tasks.append(task)
    job.status = job.WAITING
    return job
コード例 #2
0
ファイル: client.py プロジェクト: HyFrmn/Wrangler
 def __init__(self, hostname=None, port=None):
     config = config_base()
     if not port:
         port = config.getint('cattle', 'port')
     if not hostname:
         hostname = info.hostname()
     WranglerClient.__init__(self, hostname, port)
コード例 #3
0
ファイル: client.py プロジェクト: HyFrmn/Wrangler
 def __init__(self, hostname=None, port=None):
     self.config = config_base()
     if not hostname:
         hostname = self.config.get('lasso', 'hostname')
     if not port:
         port = self.config.getint('lasso', 'port')
     log.debug('Starting client to %s:%d' % (hostname, port))
     self.client = ServerProxy('http://%s:%s' % (hostname, port), allow_none=True)
コード例 #4
0
ファイル: generators.py プロジェクト: HyFrmn/Wrangler
def update_env(env):
    run_env = {}#os.environ.copy()
    env_mask = config_base().get('jobs', 'env-mask').split(':')
    for mask in env_mask:
        try:
            del env[mask]
        except KeyError:
            pass
    run_env.update(env)
    return run_env
コード例 #5
0
ファイル: server.py プロジェクト: HyFrmn/Wrangler
 def configure(self):
     self.config = config_base()
コード例 #6
0
ファイル: log.py プロジェクト: HyFrmn/Wrangler
#!/usr/bin/env python

import os


from wrangler.config import config_base
from wrangler.db.core import *

config = config_base()

class TaskLog(Base):
    __tablename__ = 'task_logs'
    
    id = Column(Integer, primary_key=True)
    task_id = Column(Integer, ForeignKey('tasks.id'))
    time = Column(Float, default=-1)
    returncode = Column(Integer, default=2)
    cattle_id = Column(Integer, ForeignKey('cattles.id'))
    cattle = relation('Cattle', backref=backref('logs'))
    stdout = Column(String(256))
    stderr = Column(String(256))

    logdir = os.path.expandvars(config.get('logging', 'task-dir'))

    def __init__(self, task, cattle):
        self.task_id = task.id
        self.cattle_id = cattle.id
        self.stdout = self._stdout_file_path()
        self.stderr = self._stderr_file_path()

    def _stdout_file_path(self):