-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merged kfsone/tradedangerous into master
- Loading branch information
Showing
10 changed files
with
541 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,183 @@ | ||
| from __future__ import absolute_import, with_statement, print_function, division, unicode_literals | ||
| from commands.commandenv import ResultRow | ||
| from commands.exceptions import * | ||
| from commands.parsing import MutuallyExclusiveGroup, ParseArgument | ||
| from formatting import RowFormat, ColumnFormat | ||
| from tradedb import TradeDB | ||
|
|
||
| ###################################################################### | ||
| # Parser config | ||
|
|
||
| help='Lists items bought/sold at a given station.' | ||
| name='market' | ||
| epilog=None | ||
| wantsTradeDB=True | ||
| arguments = [ | ||
| ParseArgument( | ||
| 'origin', | ||
| help='Station being queried.', | ||
| metavar='STATIONNAME', | ||
| type=str, | ||
| ), | ||
| ] | ||
| switches = [ | ||
| ParseArgument( | ||
| '--buying', '-B', | ||
| help='Show items station is buying', | ||
| action='store_true', | ||
| ), | ||
| ParseArgument( | ||
| '--selling', '-S', | ||
| help='Show items station is selling', | ||
| action='store_true', | ||
| ) | ||
| ] | ||
|
|
||
| ###################################################################### | ||
| # Perform query and populate result set | ||
|
|
||
|
|
||
| def render_units(units, level): | ||
| if level == 0: | ||
| return '-' | ||
| if units < 0: | ||
| return '?' | ||
| levelNames = { -1: '?', 1: 'L', 2: 'M', 3: 'H' } | ||
| return "{:n}{}".format(units, levelNames[level]) | ||
|
|
||
|
|
||
| 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 | ||
| if not buying and not selling: | ||
| raise CommandLineError( | ||
| "Please specify one or both of --buying (-B) " | ||
| "or --selling (-S)." | ||
| ) | ||
|
|
||
| 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 si.item_id, | ||
| sb.price, | ||
| sb.units, | ||
| sb.level, | ||
| JULIANDAY('now') - JULIANDAY(sb.modified), | ||
| ss.price, | ||
| ss.units, | ||
| ss.level, | ||
| JULIANDAY('now') - JULIANDAY(ss.modified) | ||
| FROM StationItem si, | ||
| StationBuying AS sb, | ||
| StationSelling AS ss | ||
| WHERE si.station_id = ? AND | ||
| sb.station_id = ? AND | ||
| ss.station_id = ? AND | ||
| sb.item_id = si.item_id AND | ||
| ss.item_id = si.item_id | ||
| """, [ | ||
| origin.ID, origin.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[item.ID] | ||
| units, level = int(next(it) or 0), int(next(it) or 0) | ||
| row.buyUnits = units | ||
| row.buyLevel = level | ||
| row.demand = render_units(units, level) | ||
| row.buyAge = float(next(it) or 0) | ||
| if buying: | ||
| hasBuy = (row.buyCr or units or level) | ||
| else: | ||
| hasBuy = False | ||
| row.sellCr = int(next(it) or 0) | ||
| row.avgSell = tdb.avgSelling[item.ID] | ||
| units, level = int(next(it) or 0), int(next(it) or 0) | ||
| row.sellUnits = units | ||
| row.sellLevel = level | ||
| row.supply = render_units(units, level) | ||
| row.sellAge = float(next(it) or 0) | ||
| if selling: | ||
| hasSell = (row.sellCr or units or level) | ||
| else: | ||
| hasSell = False | ||
| 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 | ||
|
|
||
| ####################################################################### | ||
| ## Transform result set into output | ||
|
|
||
|
|
||
| def render(results, cmdenv, tdb): | ||
| longest = max(results.rows, key=lambda row: len(row.item.name())) | ||
| longestLen = len(longest.item.name()) | ||
| longestDmd = max(results.rows, key=lambda row: len(row.demand)).demand | ||
| longestSup = max(results.rows, key=lambda row: len(row.supply)).supply | ||
| dmdLen = max(len(longestDmd), len("Demand")) | ||
| supLen = max(len(longestSup), len("Supply")) | ||
|
|
||
| showCategories = (cmdenv.detail > 0) | ||
|
|
||
| rowFmt = RowFormat() | ||
| if showCategories: | ||
| rowFmt.prefix = ' ' | ||
|
|
||
| rowFmt.addColumn('Item', '<', longestLen, | ||
| key=lambda row: row.item.name()) | ||
| if cmdenv.buying: | ||
| rowFmt.addColumn('Buying', '>', 7, 'n', | ||
| key=lambda row: row.buyCr) | ||
| if cmdenv.detail: | ||
| rowFmt.addColumn('Avg', '>', 7, 'n', | ||
| key=lambda row: row.avgBuy) | ||
| if cmdenv.detail > 1: | ||
| rowFmt.addColumn('Demand', '>', dmdLen, | ||
| key=lambda row: row.demand) | ||
| if cmdenv.detail: | ||
| rowFmt.addColumn('Age/Days', '>', 7, '.2f', | ||
| key=lambda row: row.buyAge) | ||
| if cmdenv.selling: | ||
| rowFmt.addColumn('Selling', '>', 7, 'n', | ||
| key=lambda row: row.sellCr) | ||
| if cmdenv.detail: | ||
| rowFmt.addColumn('Avg', '>', 7, 'n', | ||
| key=lambda row: row.avgSell) | ||
| rowFmt.addColumn('Supply', '>', supLen, | ||
| key=lambda row: row.supply) | ||
| if cmdenv.detail: | ||
| rowFmt.addColumn('Age/Days', '>', 7, '.2f', | ||
| key=lambda row: row.buyAge) | ||
|
|
||
| if not cmdenv.quiet: | ||
| heading, underline = rowFmt.heading() | ||
| print(heading, underline, sep='\n') | ||
|
|
||
| lastCat = None | ||
| for row in results.rows: | ||
| if showCategories and row.item.category is not lastCat: | ||
| print("+{}".format(row.item.category.name())) | ||
| lastCat = row.item.category | ||
| print(rowFmt.format(row)) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -88,6 +88,11 @@ unq:[email protected]_id,unq:[email protected]_id,unq:[email protected]_id | |
| 'APTET','Torricelli Port','Type 6' | ||
| 'APTET','Torricelli Port','Type 7' | ||
| 'APTET','Torricelli Port','Type 9' | ||
| 'ATTACATES','Buckell Ring','Cobra' | ||
| 'ATTACATES','Buckell Ring','Eagle' | ||
| 'ATTACATES','Buckell Ring','Sidewinder' | ||
| 'ATTACATES','Buckell Ring','Type 6' | ||
| 'ATTACATES','Buckell Ring','Type 7' | ||
| 'BAST','Hart Station','Cobra' | ||
| 'BAST','Hart Station','Eagle' | ||
| 'BHRITZAMENO','Feynman Terminal','Adder' | ||
|
|
||
Oops, something went wrong.