davorian
/
space-haven
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added coloured text using Rich, also better storage handling.

This commit is contained in:
Chris Davoren 2023-07-11 17:26:52 +10:00
parent 8897a97777
commit d507c1b1d1
2 changed files with 145 additions and 101 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements.txt
View File

@ -1,3 +1,8 @@
beautifulsoup4==4.12.2
colorama==0.4.6
lxml==4.9.2
markdown-it-py==3.0.0
mdurl==0.1.2
Pygments==2.15.1
rich==13.4.2
soupsieve==2.4.1

113
sheditor.py
View File

@ -1,8 +1,23 @@
import sys, argparse, copy, os, datetime
import argparse
import copy
import os
import datetime
from bs4 import BeautifulSoup
import rich
from rich.console import Console
# Only "normal storage" is used to insert requested items
normal_storage_ids = [82, 632]
storage_ids = {
82: {"name": "Small storage", "capacity": 50},
632: {"name": "Large storage", "capacity": 250},
3062: {"name": "Body storage", "capacity": 20},
}
char_skills = {
1 : "Piloting", # Likely deprecated
1: "Piloting", # Likely deprecated - now using code 22
2: "Mining",
3: "Botany",
4: "Construction",
@ -55,6 +70,7 @@ char_traits = {
"Gourmand": 1562,
}
class ItemCodeDatabase:
def __init__(self, database_filename):
@ -88,6 +104,7 @@ class StorageArea:
def __init__(self, tag):
self.tag = tag
self.type_id = None
self.items = []
self.is_abnormal_storage = False
@ -148,15 +165,24 @@ class Character:
return new_char
def print_summary(self):
print("Name: {}".format(self.name))
print()
print("Skills:")
for skill in self.skills:
print("{} : {}".format(char_skills[skill['id']], skill['level']))
print()
print("Attributes:")
for attribute in self.attributes:
print("{} : {}".format(char_attributes[attribute['id']], attribute['points']))
console = Console(highlight=False)
console.print("Name: [bright_cyan]{}[/]".format(self.name))
console.print()
console.print("Skills:")
for i, skill in enumerate(self.skills):
row_string = "{:20} [bright_cyan]{:2}[/]".format(char_skills[skill['id']], skill['level'])
if i % 2 == 0:
console.print(" [on #222222]{}[/]".format(row_string))
else:
console.print(" {}".format(row_string))
console.print()
console.print("Attributes:")
for i, attribute in enumerate(self.attributes):
row_string = "{:12} [bright_cyan]{:2}[/]".format(char_attributes[attribute['id']], attribute['points'])
if i % 2 == 0:
console.print(" [on #222222]{}[/]".format(row_string))
else:
console.print(" {}".format(row_string))
def __repr__(self):
return "{} - Skills: {} - Attributes: {}".format(self.name, repr(self.skills), repr(self.attributes))
@ -223,6 +249,7 @@ class GameData:
ship_name = ship_tag['sname']
# print(ship_name)
# Forgotten - why did I specify owner as required here - what ships don't have owners?
owner_node = ship_tag.find('settings', owner=True)
ship_owner = owner_node['owner']
ship_state = owner_node['state']
@ -233,8 +260,10 @@ class GameData:
# Step 3 - Storage area data
for inv_tag in ship_tag.find_all('feat', eatAllowed=True):
storage_area = StorageArea(inv_tag.find('inv'))
# This is a GUESS
if inv_tag.find('env'):
# 632 is the "m" id code for LARGE storage
storage_area.type_id = int(inv_tag.parent.parent['m'])
if storage_area.type_id not in normal_storage_ids: # TODO: Un-hardcode this AND take into account SMALL storage
# rich.print("[on #222222]Found abnormal storage id [#666666]{}[/][/]".format(inv_tag.parent.parent['m']))
storage_area.is_abnormal_storage = True
ship.add_storage_area(storage_area)
# Items within storage area
@ -284,7 +313,6 @@ class GameData:
attributes.append(attribute_dict)
character.set_attributes(attributes)
def writeback(self):
def replace_id(dict, old_key, new_key):
dict_copy = dict.copy()
@ -364,40 +392,53 @@ class GameData:
return
def print_detailed_character_summary(self):
console = Console(highlight=False)
for ship in self.ships:
console.print("Listing characters for ship [magenta]{}[/] (owned by [bright_cyan]{}[/], state [bright_cyan]{}[/]):".format(ship.name, ship.owner, ship.state))
if len(ship.characters) == 0:
print("Ship {} (owned by {}, state {}) has no characters, skipping...".format(ship.name, ship.owner, ship.state))
print()
console.print(" This ship has no characters, skipping...")
console.print()
console.print()
continue
print("Listing characters for ship {} (owned by {}, state {}):".format(ship.name, ship.owner, ship.state))
print('-----')
for character in ship.characters:
character.print_summary()
print('-----')
console.print('-----')
print()
console.print()
def print_detailed_item_summary(self):
console = Console(highlight=False)
for ship in self.ships:
rich.print("Inventory for ship [#aa00ff]{}[/] (owned by [bright_cyan]{}[/], state [bright_cyan]{}[/]):".format(ship.name, ship.owner, ship.state))
if len(ship.storage_areas) == 0:
print("Ship {} (owned by {}, state {}) has no storage areas, skipping...".format(ship.name, ship.owner, ship.state))
print()
console.print(" This ship has no storage areas, skipping...")
console.print()
console.print()
continue
print("Inventory for ship {} (owned by {}, state {})".format(ship.name, ship.owner, ship.state))
for index, storage_area in enumerate(ship.storage_areas):
console.print(" Storage area [bright_cyan]{}[/] (type: [bright_cyan]{}[/], abnormal storage: [bright_cyan]{}[/]):".format(index, storage_ids[storage_area.type_id]["name"], storage_area.is_abnormal_storage))
if len(storage_area.items) == 0:
print(" Storage area {} is empty.".format(index))
console.print(" This storage area is empty.")
console.print()
continue
print(" Storage area {} (abnormal storage: {}):".format(index, storage_area.is_abnormal_storage))
for item in storage_area.items:
print(" {:4}: {} - {}".format(item.code, item.name, item.quantity))
print()
print()
console.print("[bold] {:>4} {:30} {:>3}[/]".format("ID", "Name", "#"))
total_quantity = 0
for i, item in enumerate(storage_area.items):
row_string = " {:4} {:30} {:3}".format(item.code, item.name, item.quantity)
total_quantity += item.quantity
if i % 2 == 0:
console.print(" [on #222222]{}[/]".format(row_string))
else:
console.print(" {}".format(row_string))
storage_area_capacity = storage_ids[storage_area.type_id]["capacity"]
console.print(" Total quantity: [bright_cyan]{}[/]/[bright_cyan]{}[/]".format(total_quantity, storage_area_capacity))
if total_quantity > storage_area_capacity:
console.print(" [dark_orange]WARNING: Storage capacity exceeded[/]")
console.print()
console.print()
def print_summary(self):
print("Start game summary:")
@ -503,7 +544,6 @@ def inventory(soup, add_code, add_quantity):
storage_space_counter += 1
print('-----')
print('Item total summary:')
print()
for item in item_tracking.items():
@ -538,9 +578,9 @@ def list_ships(soup):
ship_owner = owner_node['owner']
ship_state = owner_node['state']
print('Ship found: {} owned by {} (state: {})'.format(ship_name, ship_owner, ship_state))
# print(settings_node)
def parse_item_file(filename):
results = []
@ -591,7 +631,8 @@ def main():
soup = BeautifulSoup(full_text, "xml")
item_code_database = ItemCodeDatabase('item_ids.txt')
print("Item code database successfully loaded")
print("Item code database successfully loaded.")
print()
game_data = GameData(soup, item_code_database)
game_data.populate()
@ -635,9 +676,6 @@ def main():
game_data.writeback()
if args.test_gamedata:
item_code_database = ItemCodeDatabase('item_ids.txt')
print("Item code database successfully loaded")
game_data = GameData(soup, item_code_database)
game_data.populate()
# game_data.add_item(1759, 100)
@ -675,5 +713,6 @@ def main():
f.write(sansfirstline)
f.close()
if __name__ == "__main__":
main()