Wow, I forgot how to write python lmao

1 files changed, 5 insertions, 4 deletions

diff --git a/src/bbss/lists.py b/src/bbss/lists.py
index 3d4779e..234d0f8 100644
--- a/src/bbss/lists.py
+++ b/src/bbss/lists.py
@@ -1,11 +1,12 @@
 from dataclasses import dataclass
 from collections.abc import Iterable, Iterator
+from typing import Optional
 import requests
 
 @dataclass(frozen=True)
 class ListFileEntry:
   entry: str
-  comment: str
+  comment: Optional[str]
 
 class ListFileIterator(Iterator[ListFileEntry]):
   def __init__(self, entries: list[ListFileEntry]):
@@ -23,13 +24,13 @@ class ListFile(Iterable[ListFileEntry]):
     self._contents = contents
     acc = []
     comment_acc = ""
-    for line in str.splitlines():
+    for line in contents.splitlines():
       if line.startswith("#"):
         continue
       elif line.startswith("##"):
-        comment_acc += line[2:].split()
+        comment_acc += line[2:].strip()
       else:
-        acc += ListFileEntry(comment = comment_acc if comment_acc else None, entry = line)
+        acc.append(ListFileEntry(comment = comment_acc if comment_acc else None, entry = line))
 
     self._entries = acc