From 2cd6514cbee5267deae882932b472bfc9737b6ed Mon Sep 17 00:00:00 2001
From: deathbybandaid <sam@deathbybandaid.net>
Date: Tue, 4 Jun 2019 07:59:07 -0400
Subject: [PATCH] attempt better load

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

diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py
index aa6aefe..019828b 100644
--- a/sopel_modules/SpiceBot/AI.py
+++ b/sopel_modules/SpiceBot/AI.py
@@ -66,26 +66,37 @@ class SpiceBot_AI():
                 dirnumber = max(self.load_commands) + 1
                 self.load_commands.append(dirnumber)
                 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"
-                        # "        <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"))
-                    )
+                # 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")
-        # for number in self.load_commands:
-        #    if number != 0:
-        #        self.aiml_kernel.respond("LOAD AIML {}").format(str(dirnumber))
-        #        os.remove(tempbrain)
+        # self.aiml_kernel.respond("LOAD AIML B")
+        for number in self.load_commands:
+            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)