def run(results, cmdenv, tdb): from commands.commandenv import ResultRow if cmdenv.lt and cmdenv.gt: if cmdenv.lt <= cmdenv.gt: raise CommandLineError("--gt must be lower than --lt") item = tdb.lookupItem(cmdenv.item) cmdenv.DEBUG0("Looking up item {} (#{})", item.name(), item.ID) avoidSystems = {s for s in cmdenv.avoidPlaces if isinstance(s, System)} avoidStations = {s for s in cmdenv.avoidPlaces if isinstance(s, Station)} results.summary = ResultRow() results.summary.item = item results.summary.avoidSystems = avoidSystems results.summary.avoidStations = avoidStations if cmdenv.detail: avgPrice = tdb.query(""" SELECT AVG(si.demand_price) FROM StationItem AS si WHERE si.item_id = ? AND si.demand_price > 0 """, [item.ID]).fetchone()[0] results.summary.avg = int(avgPrice) # Constraints tables = "StationItem AS si" constraints = [ "(item_id = {} AND demand_price > 0)".format(item.ID), ] columns = [ 'si.station_id', 'si.demand_price', 'si.demand_units', ] bindValues = [] if cmdenv.demand: constraints.append("(demand_units >= ?)") bindValues.append(cmdenv.demand) if cmdenv.lt: constraints.append("(demand_price < ?)") bindValues.append(cmdenv.lt) if cmdenv.gt: constraints.append("(demand_price > ?)") bindValues.append(cmdenv.gt) nearSystem = cmdenv.nearSystem if nearSystem: maxLy = cmdenv.maxLyPer or tdb.maxSystemLinkLy results.summary.near = nearSystem results.summary.ly = maxLy distanceFn = nearSystem.distanceTo else: distanceFn = None whereClause = ' AND '.join(constraints) stmt = """SELECT DISTINCT {columns} FROM {tables} WHERE {where}""".format( columns=','.join(columns), tables=tables, where=whereClause ) cmdenv.DEBUG0('SQL: {}', stmt) cur = tdb.query(stmt, bindValues) stationByID = tdb.stationByID padSize = cmdenv.padSize planetary = cmdenv.planetary wantNoPlanet = cmdenv.noPlanet wantBlackMarket = cmdenv.blackMarket for (stationID, priceCr, demand) in cur: station = stationByID[stationID] if padSize and not station.checkPadSize(padSize): continue if planetary and not station.checkPlanetary(planetary): continue if wantNoPlanet and station.planetary != 'N': continue if wantBlackMarket and station.blackMarket != 'Y': continue if station in avoidStations: continue if station.system in avoidSystems: continue row = ResultRow() row.station = station if distanceFn: distance = distanceFn(row.station.system) if distance > maxLy: continue row.dist = distance row.price = priceCr row.demand = demand row.age = station.itemDataAgeStr results.rows.append(row) if not results.rows: raise NoDataError("No available items found") results.summary.sort = "Price" results.rows.sort(key=lambda result: result.demand, reverse=True) results.rows.sort(key=lambda result: result.price, reverse=True) if nearSystem and not cmdenv.sortByPrice: results.summary.sort = "Dist" results.rows.sort(key=lambda result: result.dist) limit = cmdenv.limit or 0 if limit > 0: results.rows = results.rows[:limit] return results
def run(results, cmdenv, tdb): origin = cmdenv.startStation if not origin.itemCount: raise CommandLineError( "No trade data available for {}".format(origin.name()) ) buying, selling = cmdenv.buying, cmdenv.selling results.summary = ResultRow() results.summary.origin = origin results.summary.buying = cmdenv.buying results.summary.selling = cmdenv.selling tdb.getAverageSelling() tdb.getAverageBuying() cur = tdb.query(""" SELECT item_id, demand_price, demand_units, demand_level, supply_price, supply_units, supply_level, JULIANDAY('now') - JULIANDAY(modified) FROM StationItem WHERE station_id = ? """, [origin.ID]) for row in cur: it = iter(row) item = tdb.itemByID[next(it)] row = ResultRow() row.item = item row.buyCr = int(next(it) or 0) row.avgBuy = tdb.avgBuying.get(item.ID, 0) units, level = int(next(it) or 0), int(next(it) or 0) row.buyUnits = units row.buyLevel = level row.demand = render_units(units, level) if not selling: hasBuy = (row.buyCr or units or level) else: hasBuy = False row.sellCr = int(next(it) or 0) row.avgSell = tdb.avgSelling.get(item.ID, 0) units, level = int(next(it) or 0), int(next(it) or 0) row.sellUnits = units row.sellLevel = level row.supply = render_units(units, level) if not buying: hasSell = (row.sellCr or units or level) else: hasSell = False row.age = float(next(it) or 0) if hasBuy or hasSell: results.rows.append(row) if not results.rows: raise CommandLineError("No items found") results.rows.sort(key=lambda row: row.item.dbname) results.rows.sort(key=lambda row: row.item.category.dbname) return results
def run(results, cmdenv, tdb): from commands.commandenv import ResultRow if cmdenv.lt and cmdenv.gt: if cmdenv.lt <= cmdenv.gt: raise CommandLineError("--gt must be lower than --lt") item = tdb.lookupItem(cmdenv.item) cmdenv.DEBUG0("Looking up item {} (#{})", item.name(), item.ID) avoidSystems = {s for s in cmdenv.avoidPlaces if isinstance(s, System)} avoidStations = {s for s in cmdenv.avoidPlaces if isinstance(s, Station)} results.summary = ResultRow() results.summary.item = item results.summary.avoidSystems = avoidSystems results.summary.avoidStations = avoidStations if cmdenv.detail: avgPrice = tdb.query(""" SELECT AVG(si.demand_price) FROM StationItem AS si WHERE si.item_id = ? AND si.demand_price > 0 """, [item.ID]).fetchone()[0] results.summary.avg = int(avgPrice) # Constraints tables = "StationItem AS si" constraints = [ "(item_id = {} AND demand_price > 0)".format(item.ID), ] columns = [ 'si.station_id', 'si.demand_price', 'si.demand_units', ] bindValues = [] if cmdenv.demand: constraints.append("(demand_units >= ?)") bindValues.append(cmdenv.demand) if cmdenv.lt: constraints.append("(demand_price < ?)") bindValues.append(cmdenv.lt) if cmdenv.gt: constraints.append("(demand_price > ?)") bindValues.append(cmdenv.gt) nearSystem = cmdenv.nearSystem if nearSystem: maxLy = cmdenv.maxLyPer or tdb.maxSystemLinkLy results.summary.near = nearSystem results.summary.ly = maxLy distanceFn = nearSystem.distanceTo else: distanceFn = None whereClause = ' AND '.join(constraints) stmt = """SELECT DISTINCT {columns} FROM {tables} WHERE {where}""".format( columns=','.join(columns), tables=tables, where=whereClause ) cmdenv.DEBUG0('SQL: {}', stmt) cur = tdb.query(stmt, bindValues) stationByID = tdb.stationByID padSize = cmdenv.padSize wantBlackMarket = cmdenv.blackMarket for (stationID, priceCr, demand) in cur: station = stationByID[stationID] if padSize and not station.checkPadSize(padSize): continue if wantBlackMarket and station.blackMarket != 'Y': continue if station in avoidStations: continue if station.system in avoidSystems: continue row = ResultRow() row.station = station if distanceFn: distance = distanceFn(row.station.system) if distance > maxLy: continue row.dist = distance row.price = priceCr row.demand = demand row.age = station.itemDataAgeStr results.rows.append(row) if not results.rows: raise NoDataError("No available items found") results.summary.sort = "Price" results.rows.sort(key=lambda result: result.demand, reverse=True) results.rows.sort(key=lambda result: result.price, reverse=True) if nearSystem and not cmdenv.sortByPrice: results.summary.sort = "Dist" results.rows.sort(key=lambda result: result.dist) limit = cmdenv.limit or 0 if limit > 0: results.rows = results.rows[:limit] return results