def main():
# parse command line arguments
argparser = argparse.ArgumentParser(
description="Create and sync Trading212 pies holdings allocations to a shared pie or external source"
)
argparser.add_argument(
"--fetch-available-equities",
type=argparse.FileType("w"),
help="Fetch the list of available equieties to trade on Trading212 invest and save it to this file. "
"when using this option, there's no need to supply email, password or pie name.",
)
if not "--fetch-available-equities" in sys.argv:
argparser.add_argument(
"username", help="The email to log into your Trading212 account"
)
argparser.add_argument(
"password", help="The password to log into your Trading212 account"
)
argparser.add_argument(
"pie", help="The name of the pie to update (case-sensitive)"
)
argparser.add_argument(
"--from-json",
type=argparse.FileType("r"),
help="Parse the list of holdings to update from this .json file "
"with the format { [ticker]: [percentage], ... }",
)
argparser.add_argument(
"--from-csv",
type=argparse.FileType("r"),
help="Parse the list of holdings to update from this .csv file "
"with the format [ticker],[percentage] for each line",
)
argparser.add_argument(
"--from-shared-pie",
help="Parse the list of instruments to update from the URL of a shared pie",
)
argparser.add_argument(
"--substitutions",
type=argparse.FileType("r"),
default="substitutions.json",
help="Parse a list of replacement tickers from this .json file, "
"To be used when a ticker is not found. The list format is "
"{ [original ticker]: [ticker to use if original not found], ... }",
)
argparser.add_argument(
"-c",
"--await-confirm",
action="store_true",
help="Do not commit changes automatically and wait for user to confirm",
)
argparser.add_argument(
"-v", "--verbose", action="store_true", help="Increase output log verbosity"
)
args = argparser.parse_args()
# configure logging for the application
log.setLevel(logging.INFO if not args.verbose else logging.DEBUG)
rich_handler = RichHandler()
rich_handler.setFormatter(logging.Formatter(fmt="%(message)s", datefmt="[%X]"))
log.addHandler(rich_handler)
log.propagate = False
# initialize chromedriver
try:
driver = ChromeDriver()
n = Navigator(driver)
except InvalidArgumentException as e:
log.error(
f"Error initalising ChromeDriver: {e}"
+ "Is another automated Chrome window still open?"
)
sys.exit(0)
if args.fetch_available_equities:
file = args.fetch_available_equities
instruments = n.get_available_instruments()
file.write(" ".join(instruments))
log.info(f"Exported {len(instruments)} instruments to {file.name}")
sys.exit(0)
# start the application
data = {"instruments": {}}
if args.from_json:
data = json.load(args.from_json)
elif args.from_csv:
reader = csv.reader(args.from_csv)
data = {rows[0]: rows[1] for rows in reader}
elif args.from_shared_pie:
data = n.parse_shared_pie(args.from_shared_pie)
n.open_dashboard(args.username, args.password)
n.select_pie(args.pie)
current_instruments = n.get_current_instruments_tickers()
unused = [
ticker
for ticker in current_instruments
if ticker not in data.keys() or round(float(data[ticker]), 1) < 0.5
]
substitutions = json.load(args.substitutions) if args.substitutions else {}
for ticker in unused:
n.remove_instrument(ticker)
for ticker, distribution in data.items():
n.rebalance_instrument(ticker, distribution, substitutions)
n.redistribute_pie()
if not args.await_confirm:
n.commit_pie_edits(name=args.pie)
else:
input("Confirm changes and then press Enter to close the browser...")
