From 2d514ba7ca7ffc83808dd312fe5fb2d0866d25ac Mon Sep 17 00:00:00 2001
From: autocommit <autocommit>
Date: Mon, 1 Jun 2020 02:09:53 +0000
Subject: [PATCH] Leftovers?

Trying to add the leftover toots that don't reply to anything at the end.
---
 brutaldon/threadtree.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/brutaldon/threadtree.py b/brutaldon/threadtree.py
index 2ff3d4c..f0a2bef 100644
--- a/brutaldon/threadtree.py
+++ b/brutaldon/threadtree.py
@@ -33,7 +33,8 @@ def maketree(descendants):
                 yield lookup[rep], onelevel(subreps)
             else:
                 yield lookup[rep], ()
-    return onelevel(roots)
+    leftovers = set(lookup.keys()) - seen
+    return onelevel(roots), (lookup[leftover] for leftover in leftovers)
 
 # returns (status, gen[(status, gen[(status, ...), (status, ())]), ...])
 
@@ -58,8 +59,13 @@ def unmaketree(tree):
             yield OUT
 
 def build(descendants):
-    derp = tuple(maketree(descendants))
+    herp, derp = maketree(descendants)
+    derp = tuple(derp)
     pprint(("derp?", derp))
     yield IN
-    yield from unmaketree(derp)
+    yield from unmaketree(herp)
+    yield OUT
+    yield IN
+    for post in derp:
+        yield POST(derp)
     yield OUT