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):
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
def run(results, cmdenv, tdb):
if cmdenv.lt and cmdenv.gt:
if cmdenv.lt <= cmdenv.gt:
raise CommandLineError("--gt must be lower than --lt")
# Find out what we're looking for.
queries, mode = get_lookup_list(cmdenv, tdb)
cmdenv.DEBUG0("{} query: {}", mode, queries.values())
avoidSystems = {s for s in cmdenv.avoidPlaces if isinstance(s, System)}
avoidStations = {s for s in cmdenv.avoidPlaces if isinstance(s, Station)}
# Summarize
results.summary = ResultRow()
results.summary.mode = mode
results.summary.queries = queries
results.summary.oneStop = cmdenv.oneStop
results.summary.avoidSystems = avoidSystems
results.summary.avoidStations = avoidStations
# In single mode with detail enabled, add average reports.
# Thus if you're looking up "algae" or the "asp", it'll
# tell you the average/ship cost.
singleMode = len(queries) == 1
if singleMode and cmdenv.detail:
first = list(queries.values())[0]
if mode is SHIP_MODE:
results.summary.avg = first.cost
else:
avgPrice = tdb.query(
"""
SELECT AVG(si.supply_price)
FROM StationItem AS si
WHERE si.item_id = ? AND si.supply_price > 0
""", [first.ID]).fetchone()[0]
if not avgPrice:
avgPrice = 0
results.summary.avg = int(avgPrice)
# System-based search
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
oneStopMode = cmdenv.oneStop
padSize = cmdenv.padSize
planetary = cmdenv.planetary
wantNoPlanet = cmdenv.noPlanet
wantBlackMarket = cmdenv.blackMarket
stations = defaultdict(list)
stationByID = tdb.stationByID
cur = sql_query(cmdenv, tdb, queries, mode)
for (ID, stationID, price, units) 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.item = queries[ID]
row.price = price
row.units = units
row.age = station.itemDataAgeStr
if oneStopMode:
stationRows = stations[stationID]
stationRows.append(row)
if len(stationRows) >= len(queries):
results.rows.extend(stationRows)
else:
results.rows.append(row)
if not results.rows:
if oneStopMode and len(stations):
raise NoDataError("No one-stop stations found")
raise NoDataError("No available items found")
if oneStopMode and not singleMode:
results.rows.sort(key=lambda result: result.item.name())
results.rows.sort(key=lambda result: result.station.name())
if cmdenv.sortByUnits:
results.summary.sort = "units"
results.rows.sort(key=lambda result: result.price)
results.rows.sort(key=lambda result: result.units, reverse=True)
else:
if not oneStopMode:
results.summary.sort = "Price"
results.rows.sort(key=lambda result: result.units, reverse=True)
results.rows.sort(key=lambda result: result.price)
if nearSystem and not cmdenv.sortByPrice:
results.summary.sort = "Ly"
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):
if cmdenv.lt and cmdenv.gt:
if cmdenv.lt <= cmdenv.gt:
raise CommandLineError("--gt must be lower than --lt")
# Find out what we're looking for.
queries, mode = get_lookup_list(cmdenv, tdb)
cmdenv.DEBUG0("{} query: {}", mode, queries.values())
avoidSystems = {s for s in cmdenv.avoidPlaces if isinstance(s, System)}
avoidStations = {s for s in cmdenv.avoidPlaces if isinstance(s, Station)}
# Summarize
results.summary = ResultRow()
results.summary.mode = mode
results.summary.queries = queries
results.summary.oneStop = cmdenv.oneStop
results.summary.avoidSystems = avoidSystems
results.summary.avoidStations = avoidStations
# In single mode with detail enabled, add average reports.
# Thus if you're looking up "algae" or the "asp", it'll
# tell you the average/ship cost.
singleMode = len(queries) == 1
if singleMode and cmdenv.detail:
first = list(queries.values())[0]
if mode is SHIP_MODE:
results.summary.avg = first.cost
else:
avgPrice = tdb.query("""
SELECT AVG(si.supply_price)
FROM StationItem AS si
WHERE si.item_id = ? AND si.supply_price > 0
""", [first.ID]).fetchone()[0]
results.summary.avg = int(avgPrice)
# System-based search
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
oneStopMode = cmdenv.oneStop
padSize = cmdenv.padSize
wantBlackMarket = cmdenv.blackMarket
stations = defaultdict(list)
stationByID = tdb.stationByID
cur = sql_query(cmdenv, tdb, queries, mode)
for (ID, stationID, price, units) 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.item = queries[ID]
row.price = price
row.units = units
row.age = station.itemDataAgeStr
if oneStopMode:
stationRows = stations[stationID]
stationRows.append(row)
if len(stationRows) >= len(queries):
results.rows.extend(stationRows)
else:
results.rows.append(row)
if not results.rows:
if oneStopMode and len(stations):
raise NoDataError("No one-stop stations found")
raise NoDataError("No available items found")
if oneStopMode and not singleMode:
results.rows.sort(key=lambda result: result.item.name())
results.rows.sort(key=lambda result: result.station.name())
if cmdenv.sortByUnits:
results.summary.sort = "units"
results.rows.sort(key=lambda result: result.price)
results.rows.sort(key=lambda result: result.units, reverse=True)
else:
if not oneStopMode:
results.summary.sort = "Price"
results.rows.sort(key=lambda result: result.units, reverse=True)
results.rows.sort(key=lambda result: result.price)
if nearSystem and not cmdenv.sortByPrice:
results.summary.sort = "Ly"
results.rows.sort(key=lambda result: result.dist)
limit = cmdenv.limit or 0
if limit > 0:
results.rows = results.rows[:limit]
return results