Refactor; memory bug in library not present anymore

.gitignore

@@ -1 +1,3 @@
 config.yaml
+__pycache__
+pics/

flake.nix

@@ -24,7 +24,7 @@
             FONTCONFIG_FILE = fontsdotconf;
           };
           buildPhase = "true";
-          installPhase = "mkdir -p $out/app; install -t $out/app *.py";
+          installPhase = "mkdir -p $out/app; install -t $out/app src/**/*.py; ";
           buildInputs = [
             (pkgs.python3.withPackages(ppkgs: [
               ppkgs.recurring-ical-events

main.py

@@ -0,0 +1,26 @@
+from PIL import Image, ImageDraw
+import locale, yaml
+from src.drawerlist import drawers
+
+locale.setlocale(locale.LC_ALL, "")
+
+def main():
+    black_img = Image.new("1", (640, 384), 0xFF)
+    red_img = Image.new("1", (640, 384), 0xFF)
+
+    black_draw = ImageDraw.Draw(black_img)
+    red_draw = ImageDraw.Draw(red_img)
+
+    config = yaml.load(open('config.yaml'), yaml.Loader)
+
+    for drawer in drawers:
+        drawer().draw(red_draw, black_draw, config)
+
+    # Save to file
+    with open("pics/black.bmp", "wb") as f:
+        black_img.save(f, "BMP")
+    with open("pics/red.bmp", "wb") as f:
+        red_img.save(f, "BMP")
+
+if __name__ == "__main__":
+    main()

src/drawer.py

@@ -0,0 +1,8 @@
+class Drawer:
+    factor = 1
+    def __init__(self) -> None:
+        pass
+    def draw(self, red_draw, black_draw, config):
+        pass
+    def f(self, i):
+        return int(self.factor * i)