ED/EDMarketConnector
1
0
Fork 0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-04-15 00:30:33 +03:00
Code Issues Projects Releases Wiki Activity

autoformatted code

Browse Source
This commit is contained in:
A_D 2020-10-02 09:32:59 +02:00 committed by Athanasius
parent c13caa22ef
commit a25ea9dfa9
1 changed files with 147 additions and 133 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

280
stats.py
View File

@ -1,7 +1,4 @@
import csv import csv
import time
from collections import OrderedDict
from functools import partial
from sys import platform from sys import platform
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
@ -20,7 +17,7 @@ from monitor import monitor
if TYPE_CHECKING: if TYPE_CHECKING:
def _(x: str) -> str: ... def _(x: str) -> str: ...
if platform=='win32': if platform == 'win32':
import ctypes import ctypes
from ctypes.wintypes import * from ctypes.wintypes import *
try: try:
@ -32,110 +29,125 @@ if platform=='win32':
GetParent.argtypes = [HWND] GetParent.argtypes = [HWND]
GetWindowRect = ctypes.windll.user32.GetWindowRect GetWindowRect = ctypes.windll.user32.GetWindowRect
GetWindowRect.argtypes = [HWND, ctypes.POINTER(RECT)] GetWindowRect.argtypes = [HWND, ctypes.POINTER(RECT)]
except: # Not supported under Wine 4.0 except: # Not supported under Wine 4.0
CalculatePopupWindowPosition = None CalculatePopupWindowPosition = None
def status(data): def status(data):
# StatsResults assumes these three things are first # StatsResults assumes these three things are first
res = [ [_('Cmdr'), data['commander']['name']], res = [
[_('Balance'), str(data['commander'].get('credits', 0))], # Cmdr stats [_('Cmdr'), data['commander']['name']],
[_('Loan'), str(data['commander'].get('debt', 0))], # Cmdr stats [_('Balance'), str(data['commander'].get('credits', 0))], # Cmdr stats
[_('Loan'), str(data['commander'].get('debt', 0))], # Cmdr stats
] ]
RANKS = [ # in output order RANKS = [ # in output order
(_('Combat') , 'combat'), # Ranking (_('Combat'), 'combat'), # Ranking
(_('Trade') , 'trade'), # Ranking (_('Trade'), 'trade'), # Ranking
(_('Explorer') , 'explore'), # Ranking (_('Explorer'), 'explore'), # Ranking
(_('CQC') , 'cqc'), # Ranking (_('CQC'), 'cqc'), # Ranking
(_('Federation') , 'federation'), # Ranking (_('Federation'), 'federation'), # Ranking
(_('Empire') , 'empire'), # Ranking (_('Empire'), 'empire'), # Ranking
(_('Powerplay') , 'power'), # Ranking (_('Powerplay'), 'power'), # Ranking
# ??? , 'crime'), # Ranking # ??? , 'crime'), # Ranking
# ??? , 'service'), # Ranking # ??? , 'service'), # Ranking
] ]
RANK_NAMES = { RANK_NAMES = {
# http://elite-dangerous.wikia.com/wiki/Pilots_Federation#Ranks # http://elite-dangerous.wikia.com/wiki/Pilots_Federation#Ranks
'combat' : [_('Harmless'), # Combat rank 'combat': [
_('Mostly Harmless'), # Combat rank _('Harmless'), # Combat rank
_('Novice'), # Combat rank _('Mostly Harmless'), # Combat rank
_('Competent'), # Combat rank _('Novice'), # Combat rank
_('Expert'), # Combat rank _('Competent'), # Combat rank
_('Master'), # Combat rank _('Expert'), # Combat rank
_('Dangerous'), # Combat rank _('Master'), # Combat rank
_('Deadly'), # Combat rank _('Dangerous'), # Combat rank
_('Elite')], # Top rank _('Deadly'), # Combat rank
'trade' : [_('Penniless'), # Trade rank _('Elite'), # Top rank
_('Mostly Penniless'), # Trade rank ],
_('Peddler'), # Trade rank 'trade': [
_('Dealer'), # Trade rank _('Penniless'), # Trade rank
_('Merchant'), # Trade rank _('Mostly Penniless'), # Trade rank
_('Broker'), # Trade rank _('Peddler'), # Trade rank
_('Entrepreneur'), # Trade rank _('Dealer'), # Trade rank
_('Tycoon'), # Trade rank _('Merchant'), # Trade rank
_('Elite')], # Top rank _('Broker'), # Trade rank
'explore' : [_('Aimless'), # Explorer rank _('Entrepreneur'), # Trade rank
_('Mostly Aimless'), # Explorer rank _('Tycoon'), # Trade rank
_('Scout'), # Explorer rank _('Elite') # Top rank
_('Surveyor'), # Explorer rank ],
_('Trailblazer'), # Explorer rank 'explore': [
_('Pathfinder'), # Explorer rank _('Aimless'), # Explorer rank
_('Ranger'), # Explorer rank _('Mostly Aimless'), # Explorer rank
_('Pioneer'), # Explorer rank _('Scout'), # Explorer rank
_('Elite')], # Top rank _('Surveyor'), # Explorer rank
'cqc' : [_('Helpless'), # CQC rank _('Trailblazer'), # Explorer rank
_('Mostly Helpless'), # CQC rank _('Pathfinder'), # Explorer rank
_('Amateur'), # CQC rank _('Ranger'), # Explorer rank
_('Semi Professional'), # CQC rank _('Pioneer'), # Explorer rank
_('Professional'), # CQC rank _('Elite') # Top rank
_('Champion'), # CQC rank ],
_('Hero'), # CQC rank 'cqc': [
_('Gladiator'), # CQC rank _('Helpless'), # CQC rank
_('Elite')], # Top rank _('Mostly Helpless'), # CQC rank
_('Amateur'), # CQC rank
_('Semi Professional'), # CQC rank
_('Professional'), # CQC rank
_('Champion'), # CQC rank
_('Hero'), # CQC rank
_('Gladiator'), # CQC rank
_('Elite') # Top rank
],
# http://elite-dangerous.wikia.com/wiki/Federation#Ranks # http://elite-dangerous.wikia.com/wiki/Federation#Ranks
'federation' : [_('None'), # No rank 'federation': [
_('Recruit'), # Federation rank _('None'), # No rank
_('Cadet'), # Federation rank _('Recruit'), # Federation rank
_('Midshipman'), # Federation rank _('Cadet'), # Federation rank
_('Petty Officer'), # Federation rank _('Midshipman'), # Federation rank
_('Chief Petty Officer'), # Federation rank _('Petty Officer'), # Federation rank
_('Warrant Officer'), # Federation rank _('Chief Petty Officer'), # Federation rank
_('Ensign'), # Federation rank _('Warrant Officer'), # Federation rank
_('Lieutenant'), # Federation rank _('Ensign'), # Federation rank
_('Lieutenant Commander'), # Federation rank _('Lieutenant'), # Federation rank
_('Post Commander'), # Federation rank _('Lieutenant Commander'), # Federation rank
_('Post Captain'), # Federation rank _('Post Commander'), # Federation rank
_('Rear Admiral'), # Federation rank _('Post Captain'), # Federation rank
_('Vice Admiral'), # Federation rank _('Rear Admiral'), # Federation rank
_('Admiral')], # Federation rank _('Vice Admiral'), # Federation rank
_('Admiral') # Federation rank
],
# http://elite-dangerous.wikia.com/wiki/Empire#Ranks # http://elite-dangerous.wikia.com/wiki/Empire#Ranks
'empire' : [_('None'), # No rank 'empire': [
_('Outsider'), # Empire rank _('None'), # No rank
_('Serf'), # Empire rank _('Outsider'), # Empire rank
_('Master'), # Empire rank _('Serf'), # Empire rank
_('Squire'), # Empire rank _('Master'), # Empire rank
_('Knight'), # Empire rank _('Squire'), # Empire rank
_('Lord'), # Empire rank _('Knight'), # Empire rank
_('Baron'), # Empire rank _('Lord'), # Empire rank
_('Viscount'), # Empire rank _('Baron'), # Empire rank
_('Count'), # Empire rank _('Viscount'), # Empire rank
_('Earl'), # Empire rank _('Count'), # Empire rank
_('Marquis'), # Empire rank _('Earl'), # Empire rank
_('Duke'), # Empire rank _('Marquis'), # Empire rank
_('Prince'), # Empire rank _('Duke'), # Empire rank
_('King')], # Empire rank _('Prince'), # Empire rank
_('King') # Empire rank
],
# http://elite-dangerous.wikia.com/wiki/Ratings # http://elite-dangerous.wikia.com/wiki/Ratings
'power' : [_('None'), # No rank 'power': [
_('Rating 1'), # Power rank _('None'), # No rank
_('Rating 2'), # Power rank _('Rating 1'), # Power rank
_('Rating 3'), # Power rank _('Rating 2'), # Power rank
_('Rating 4'), # Power rank _('Rating 3'), # Power rank
_('Rating 5')], # Power rank _('Rating 4'), # Power rank
_('Rating 5') # Power rank
],
} }
ranks = data['commander'].get('rank', {}) ranks = data['commander'].get('rank', {})
@ -144,8 +156,9 @@ def status(data):
names = RANK_NAMES[thing] names = RANK_NAMES[thing]
if isinstance(rank, int): if isinstance(rank, int):
res.append([title, rank < len(names) and names[rank] or ('Rank %d' % rank)]) res.append([title, rank < len(names) and names[rank] or ('Rank %d' % rank)])
else: else:
res.append([title, _('None')]) # No rank res.append([title, _('None')]) # No rank
return res return res
@ -163,27 +176,26 @@ def ships(data):
current = data['commander'].get('currentShipId') current = data['commander'].get('currentShipId')
if isinstance(current, int) and current < len(ships) and ships[current]: if isinstance(current, int) and current < len(ships) and ships[current]:
ships.insert(0, ships.pop(current)) # Put current ship first ships.insert(0, ships.pop(current)) # Put current ship first
if not data['commander'].get('docked'): if not data['commander'].get('docked'):
# Set current system, not last docked # Set current system, not last docked
return ( return (
[( [(
str(ships[0]['id']), str(ships[0]['id']),
ship_map.get(ships[0]['name'].lower(), ship_map.get(ships[0]['name'].lower(),
ships[0]['name']), ships[0]['name']),
ships[0].get('shipName', ''), ships[0].get('shipName', ''),
data['lastSystem']['name'], data['lastSystem']['name'],
'', str(ships[0]['value']['total']) '', str(ships[0]['value']['total'])
)] + )] +
[( [(
str(ship['id']), ship_map.get(ship['name'].lower(), str(ship['id']), ship_map.get(ship['name'].lower(), ship['name']),
ship['name']), ship.get('shipName', ''),
ship.get('shipName', ''), ship['starsystem']['name'],
ship['starsystem']['name'], ship['station']['name'],
ship['station']['name'], str(ship['value']['total'])) for ship in ships[1:] if ship
str(ship['value']['total'])) for ship in ships[1:] if ship ])
])
return [ return [
( (
@ -205,7 +217,6 @@ def export_ships(data, filename):
class StatsDialog(): class StatsDialog():
def __init__(self, app): def __init__(self, app):
self.parent = app.w self.parent = app.w
self.status = app.status self.status = app.status
@ -226,23 +237,24 @@ class StatsDialog():
return return
except Exception as e: except Exception as e:
if __debug__: print_exc() if __debug__:
print_exc()
self.status['text'] = str(e) self.status['text'] = str(e)
return return
if not data.get('commander') or not data['commander'].get('name','').strip(): if not data.get('commander') or not data['commander'].get('name', '').strip():
self.status['text'] = _("Who are you?!") # Shouldn't happen self.status['text'] = _("Who are you?!") # Shouldn't happen
elif ( elif (
not data.get('lastSystem') not data.get('lastSystem')
or not data['lastSystem'].get('name','').strip() or not data['lastSystem'].get('name', '').strip()
or not data.get('lastStarport') or not data.get('lastStarport')
or not data['lastStarport'].get('name','').strip() or not data['lastStarport'].get('name', '').strip()
): ):
self.status['text'] = _("Where are you?!") # Shouldn't happen self.status['text'] = _("Where are you?!") # Shouldn't happen
elif not data.get('ship') or not data['ship'].get('modules') or not data['ship'].get('name','').strip(): elif not data.get('ship') or not data['ship'].get('modules') or not data['ship'].get('name', '').strip():
self.status['text'] = _("What are you flying?!") # Shouldn't happen self.status['text'] = _("What are you flying?!") # Shouldn't happen
else: else:
self.status['text'] = '' self.status['text'] = ''
@ -250,27 +262,27 @@ class StatsDialog():
class StatsResults(tk.Toplevel): class StatsResults(tk.Toplevel):
def __init__(self, parent, data): def __init__(self, parent, data):
tk.Toplevel.__init__(self, parent) tk.Toplevel.__init__(self, parent)
self.parent = parent self.parent = parent
stats = status(data) stats = status(data)
self.title(' '.join(stats[0])) # assumes first thing is player name self.title(' '.join(stats[0])) # assumes first thing is player name
if parent.winfo_viewable(): if parent.winfo_viewable():
self.transient(parent) self.transient(parent)
# position over parent # position over parent
if platform!='darwin' or parent.winfo_rooty()>0: # http://core.tcl.tk/tk/tktview/c84f660833546b1b84e7 if platform != 'darwin' or parent.winfo_rooty() > 0: # http://core.tcl.tk/tk/tktview/c84f660833546b1b84e7
self.geometry("+%d+%d" % (parent.winfo_rootx(), parent.winfo_rooty())) self.geometry("+%d+%d" % (parent.winfo_rootx(), parent.winfo_rooty()))
# remove decoration # remove decoration
self.resizable(tk.FALSE, tk.FALSE) self.resizable(tk.FALSE, tk.FALSE)
if platform=='win32': if platform == 'win32':
self.attributes('-toolwindow', tk.TRUE) self.attributes('-toolwindow', tk.TRUE)
elif platform=='darwin':
elif platform == 'darwin':
# http://wiki.tcl.tk/13428 # http://wiki.tcl.tk/13428
parent.call('tk::unsupported::MacWindowStyle', 'style', self, 'utility') parent.call('tk::unsupported::MacWindowStyle', 'style', self, 'utility')
@ -281,7 +293,7 @@ class StatsResults(tk.Toplevel):
page = self.addpage(notebook) page = self.addpage(notebook)
for thing in stats[1:3]: for thing in stats[1:3]:
self.addpagerow(page, [thing[0], self.credits(int(thing[1]))]) # assumes things two and three are money self.addpagerow(page, [thing[0], self.credits(int(thing[1]))]) # assumes things two and three are money
for thing in stats[3:]: for thing in stats[3:]:
self.addpagerow(page, thing) self.addpagerow(page, thing)
@ -290,24 +302,25 @@ class StatsResults(tk.Toplevel):
notebook.add(page, text=_('Status')) # Status dialog title notebook.add(page, text=_('Status')) # Status dialog title
page = self.addpage(notebook, [ page = self.addpage(notebook, [
_('Ship'), # Status dialog subtitle _('Ship'), # Status dialog subtitle
'', '',
_('System'), # Main window _('System'), # Main window
_('Station'), # Status dialog subtitle _('Station'), # Status dialog subtitle
_('Value'), # Status dialog subtitle - CR value of ship _('Value'), # Status dialog subtitle - CR value of ship
]) ])
shiplist = ships(data) shiplist = ships(data)
for thing in shiplist: for thing in shiplist:
self.addpagerow(page, list(thing[1:-1]) + [self.credits(int(thing[-1]))]) # skip id, last item is money self.addpagerow(page, list(thing[1:-1]) + [self.credits(int(thing[-1]))]) # skip id, last item is money
ttk.Frame(page).grid(pady=5) # bottom spacer ttk.Frame(page).grid(pady=5) # bottom spacer
notebook.add(page, text=_('Ships')) # Status dialog title notebook.add(page, text=_('Ships')) # Status dialog title
if platform!='darwin': if platform != 'darwin':
buttonframe = ttk.Frame(frame) buttonframe = ttk.Frame(frame)
buttonframe.grid(padx=10, pady=(0,10), sticky=tk.NSEW) buttonframe.grid(padx=10, pady=(0, 10), sticky=tk.NSEW)
buttonframe.columnconfigure(0, weight=1) buttonframe.columnconfigure(0, weight=1)
ttk.Label(buttonframe).grid(row=0, column=0) # spacer ttk.Label(buttonframe).grid(row=0, column=0) # spacer
ttk.Button(buttonframe, text='OK', command=self.destroy).grid(row=0, column=1, sticky=tk.E) ttk.Button(buttonframe, text='OK', command=self.destroy).grid(row=0, column=1, sticky=tk.E)
# wait for window to appear on screen before calling grab_set # wait for window to appear on screen before calling grab_set
@ -318,9 +331,11 @@ class StatsResults(tk.Toplevel):
if platform == 'win32' and CalculatePopupWindowPosition: if platform == 'win32' and CalculatePopupWindowPosition:
position = RECT() position = RECT()
GetWindowRect(GetParent(self.winfo_id()), position) GetWindowRect(GetParent(self.winfo_id()), position)
if CalculatePopupWindowPosition(POINT(parent.winfo_rootx(), parent.winfo_rooty()), if CalculatePopupWindowPosition(
SIZE(position.right - position.left, position.bottom - position.top), POINT(parent.winfo_rootx(), parent.winfo_rooty()),
0x10000, None, position): SIZE(position.right - position.left, position.bottom - position.top),
0x10000, None, position
):
self.geometry("+%d+%d" % (position.left, position.top)) self.geometry("+%d+%d" % (position.left, position.top))
def addpage(self, parent, header=[], align=None): def addpage(self, parent, header=[], align=None):
@ -346,7 +361,7 @@ class StatsResults(tk.Toplevel):
label.grid(padx=10, sticky=tk.W) label.grid(padx=10, sticky=tk.W)
row = parent.grid_size()[1]-1 row = parent.grid_size()[1]-1
elif align is None and i == len(content) - 1: # Assumes last column right justified if unspecified elif align is None and i == len(content) - 1: # Assumes last column right justified if unspecified
label.grid(row=row, column=i, padx=10, sticky=tk.E) label.grid(row=row, column=i, padx=10, sticky=tk.E)
else: else:
@ -354,4 +369,3 @@ class StatsResults(tk.Toplevel):
def credits(self, value): def credits(self, value):
return Locale.stringFromNumber(value, 0) + ' Cr' return Locale.stringFromNumber(value, 0) + ' Cr'