Ejemplo n.º 1
0
 def __init__(self, state, x_next, depth, x_depth, directed, neighbour):
     super(DistanceControl, self).__init__(directed)
     self.__x_next = x_next
     self.__depth = depth
     self.__x_depth = x_depth
     self.__neighbour = neighbour
     self.__max_weight = state.session.query(max_(Graph.weight)).one()[0]
Ejemplo n.º 2
0
 def __init__(self, state, x_next, depth, x_depth, directed, neighbour):
     super(WeightedControl, self).__init__(directed)
     self.__x_next = x_next
     self.__depth = depth
     self.__x_depth = x_depth
     self.__neighbour = neighbour
     self.__max_weight = state.session.query(max_(Graph.weight)).one()[0]
Ejemplo n.º 3
0
    def add_calculation(self, proposal_id, calculator_id, mode, version,
                        input_, output, calc_version, title):
        with self._transaction() as conn:
            calculation_alias = calculation.alias()

            result = conn.execute(calculation.insert().values({
                calculation.c.proposal_id:
                proposal_id,
                calculation.c.sort_order:
                select([
                    coalesce(max_(calculation_alias.c.sort_order), 0) + 1
                ]).where(calculation_alias.c.proposal_id == proposal_id),
                calculation.c.calculator_id:
                calculator_id,
                calculation.c.mode:
                mode,
                calculation.c.version:
                version,
                calculation.c.input:
                input_,
                calculation.c.output:
                output,
                calculation.c.date_run:
                datetime.utcnow(),
                calculation.c.calc_version:
                calc_version,
                calculation.c.title:
                title,
            }))

            return result.inserted_primary_key[0]
Ejemplo n.º 4
0
def collect(session, zero):
    max_weight = session.query(max_(Graph.weight)).one()[0]
    zero *= max_weight
    LOG.info('Collecting all nodes, edges with weights over {0:7.1f} (max {1:7.1f}).'.format(zero, max_weight))
    edges =  set(session.query(Graph).filter(Graph.weight > zero).all())
    nodes = set(edge.from_ for edge in edges)
    nodes.update(edge.to_ for edge in edges)
    LOG.info('Collected {0}/{1} nodes and {2}/{3} edges.'.format(
            len(nodes), session.query(LastFmArtist).count(), 
            len(edges), session.query(Graph).count()))
    return (nodes, edges, zero)
Ejemplo n.º 5
0
def collect(session, zero):
    max_weight = session.query(max_(Graph.weight)).one()[0]
    zero *= max_weight
    LOG.info(
        'Collecting all nodes, edges with weights over {0:7.1f} (max {1:7.1f}).'
        .format(zero, max_weight))
    edges = set(session.query(Graph).filter(Graph.weight > zero).all())
    nodes = set(edge.from_ for edge in edges)
    nodes.update(edge.to_ for edge in edges)
    LOG.info('Collected {0}/{1} nodes and {2}/{3} edges.'.format(
        len(nodes),
        session.query(LastFmArtist).count(), len(edges),
        session.query(Graph).count()))
    return (nodes, edges, zero)
Ejemplo n.º 6
0
    def add_calculation(self, proposal_id, calculator_id,
                        mode, version, input_, output, calc_version, title):
        with self._transaction() as conn:
            calculation_alias = calculation.alias()

            result = conn.execute(calculation.insert().values({
                calculation.c.proposal_id: proposal_id,
                calculation.c.sort_order: select(
                    [coalesce(max_(calculation_alias.c.sort_order), 0) + 1]
                    ).where(calculation_alias.c.proposal_id == proposal_id),
                calculation.c.calculator_id: calculator_id,
                calculation.c.mode: mode,
                calculation.c.version: version,
                calculation.c.input: input_,
                calculation.c.output: output,
                calculation.c.date_run: datetime.utcnow(),
                calculation.c.calc_version: calc_version,
                calculation.c.title: title,
            }))

            return result.inserted_primary_key[0]