From dbb2700eac87d0bb89d1b36ece5d4f0dd2f63572 Mon Sep 17 00:00:00 2001
From: deathbybandaid <sam@deathbybandaid.net>
Date: Tue, 4 Jun 2019 08:19:08 -0400
Subject: [PATCH] test

---
 sopel_modules/SpiceBot/AI.py | 49 +++++++++++++++++++++++++-----------
 1 file changed, 34 insertions(+), 15 deletions(-)

diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py
index 8c8afb2..d1507d6 100644
--- a/sopel_modules/SpiceBot/AI.py
+++ b/sopel_modules/SpiceBot/AI.py
@@ -22,8 +22,8 @@ class SpiceBot_AI_MainSection(StaticSection):
 class SpiceBot_AI():
 
     def __init__(self):
-        self.load_commands = [0]
-        self.braindirs = {}
+        self.load_commands = {0: None}
+        self.braindirs = []
         self.dict = {
                     "counts": 0,
                     "failcounts": 0,
@@ -63,21 +63,40 @@ class SpiceBot_AI():
         for braindir in braindirs:
             if braindir not in self.braindirs:
                 self.braindirs.append(braindir)
+                dirnumber = max(self.load_commands.keys()) + 1
+                self.load_commands[dirnumber] = braindir
                 tempbrain = tempfile.mkstemp()[1]
-                with open(tempbrain, 'w') as fileo:
-                    fileo.write(
-                        "<aiml version='1.0.1' encoding='UTF-8'>"
-                        "    <!-- std-startup.xml -->\n"
-                        "    <category>\n"
-                        "        <pattern>LOAD AIML B</pattern>\n"
-                        "        <template>\n"
-                        "            <learn>{}</learn>\n"
-                        "        </template>\n"
-                        "    </category>\n"
-                        "</aiml>".format(os.path.join(braindir, "*.aiml"))
-                    )
+                # with open(tempbrain, 'w') as fileo:
+                #    fileo.write(
+                #        "<aiml version='1.0.1' encoding='UTF-8'>"
+                #        "    <!-- std-startup.xml -->\n"
+                #        "    <category>\n"
+                #        "        <pattern>LOAD AIML B</pattern>\n"
+                #        # "        <pattern>LOAD AIML {}</pattern>\n"
+                #        "        <template>\n"
+                #        "            <learn>{}</learn>\n"
+                #        "        </template>\n"
+                #        "    </category>\n"
+                #        # "</aiml>".format(str(dirnumber), os.path.join(braindir, "*"))
+                #        "</aiml>".format(os.path.join(braindir, "*.aiml"))
+                #    )
+                brainfile = open(tempbrain, 'a')
+                brainfile.write("<aiml version='1.0.1' encoding='UTF-8'>")
+                brainfile.write("    <!-- std-startup.xml -->\n")
+                brainfile.write("    <category>\n")
+                brainfile.write("        <pattern>LOAD AIML {}</pattern>\n".format(str(dirnumber)))
+                brainfile.write("        <template>\n")
+                brainfile.write("            <learn>{}</learn>\n".format(os.path.join(braindir, "*.aiml")))
+                brainfile.write("        </template>\n")
+                brainfile.write("    </category>\n")
+                brainfile.write("</aiml>")
+                brainfile.close()
                 self.aiml_kernel.learn(tempbrain)
-        self.aiml_kernel.respond("LOAD AIML B")
+        # self.aiml_kernel.respond("LOAD AIML B")
+        for number in self.load_commands.keys():
+            if number != 0:
+                self.aiml_kernel.respond("LOAD AIML {}").format(str(dirnumber))
+                os.remove(tempbrain)
 
     def on_message(self, bot, trigger, message):
         nick = Identifier(trigger.nick)