Added user config yaml

Author: AlgorithmEnigma

.gitignore

@@ -3,4 +3,5 @@
 .vscode
 /spacesay/__pycache__
 /dist
-poetry.lock
+poetry.lock
+config.yaml

pyproject.toml

@@ -11,8 +11,8 @@ spacesay = "spacesay.main:app"
 [tool.poetry.dependencies]
 python = "^3.10"
 typer = {extras = ["all"], version = "^0.7.0"}
-python-dotenv = "^0.21.1"
 requests = "^2.28.2"
+pyyaml = "^6.0"
 
 
 [build-system]

spacesay/location.py

@@ -4,7 +4,7 @@
 from datetime import date
 
 
-def iss_location() -> list[str]:
+def iss_location(username: str) -> list[str]:
     # Request ISS location from open-notify
     try:
         r = requests.get("http://api.open-notify.org/iss-now.json").json()
@@ -30,7 +30,7 @@ def iss_location() -> list[str]:
         querystring_geonames = {
             "lat": {lat},
             "lng": {lng},
-            "username": os.environ.get("GEONAMES_USERNAME"),
+            "username": username,
         }
 
         responce_geonames = requests.request(

spacesay/main.py

@@ -1,12 +1,44 @@
 import typer
+import rich
+import yaml
 from typing import Optional
-from dotenv import load_dotenv
+
 
 from spacesay.location import iss_location
 from spacesay.speech import speech_bubble
 
 
-load_dotenv()  # Export dotenv into enviroment
+# Create or open a YAML config file and set username
+def set_username() -> str:
+    try:
+        stream = open("config.yaml", "w")
+        rich.print(
+            "Username not set, please visit [bold magenta]https://www.geonames.org/login[/bold magenta] to sign up and get a username."
+        )
+        username = input("Please enter your username: ")
+        yaml.dump({"username": username}, stream)
+        return username
+    except:
+        print("Failed to create config")
+        raise FileNotFoundError
+
+
+username = ""
+
+try:
+    stream = open("config.yaml", "r")
+    config = yaml.safe_load(stream)
+    try:
+        if config["username"] is None:
+            username = set_username()
+        else:
+            username = config["username"]
+    except:
+        username = set_username()
+
+except:
+    print("Failed to open config, creating config file")
+    username: str = set_username()
 
 
 astronaut = """
@@ -36,7 +68,7 @@
 @app.command()
 def main(text: Optional[str] = typer.Argument(None)):
     if text is None:
-        location_time = iss_location()
+        location_time = iss_location(username)
         speech_bubble(location_time)
 
     else: