Deleting files and completing event generation.
[smartthings-infrastructure.git] / Extractor / ExtractorScript.py
index 40fa93d98fec8e43972fa7743497f3cd72facfac..e2edcad4f22a8d81c585ffd1532c44171edcf826 100644 (file)
@@ -58,7 +58,7 @@ def ExtractFunctions(F, appName):
                        Temp = GetToken(F)
                        #If counter >= 5 that means it is not found, so it must be appTouch
                        if (counter >= 5):
-                               Temp = "appTouch"
+                               Temp = "Touched"
                        eventMap.append(Temp)
                Temp = GetToken(F)
                
@@ -69,38 +69,90 @@ def ExtractEvents(extractedEvents):
        extractedEvents.write("while(counter > 0) {\n")
        extractedEvents.write("\tdef eventNumber = random.nextInt(%d)\n" % (len(eventMap) - 1))
        extractedEvents.write("\tswitch(eventNumber) {\n")
+       eventCounter = 0;
        for i in range(len(eventMap)):
-               extractedEvents.write("\t\tcase %d:\n" % i)
+               extractedEvents.write("\t\tcase %d:\n" % eventCounter)
                if eventMap[i] == "lock":
-                       event = open("eventSimulator/eventLocked.groovy", "r")
+                       event = open("eventSimulator/lockEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
                elif eventMap[i] == "unlock":
-                       event = open("eventSimulator/eventUnlocked.groovy", "r")
+                       event = open("eventSimulator/unlockEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
                elif eventMap[i] == "contact.open":
-                       event = open("eventSimulator/eventContactOpen.groovy", "r")
+                       event = open("eventSimulator/contactOpenEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
                elif eventMap[i] == "contact.closed":
-                       event = open("eventSimulator/eventContactClosed.groovy", "r")
+                       event = open("eventSimulator/contactClosedEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+               elif eventMap[i] == "switch.on":
+                       event = open("eventSimulator/switchOnEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+               elif eventMap[i] == "switch.off":
+                       event = open("eventSimulator/switchOffEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
                elif eventMap[i] == "nfcTouch":
-                       event = open("eventSimulator/eventNfcTouch.groovy", "r")
+                       event = open("eventSimulator/nfcTouchEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
-               elif eventMap[i] == "appTouch":
-                       event = open("eventSimulator/eventAppTouch.groovy", "r")
+               elif eventMap[i] == "Touched":
+                       event = open("eventSimulator/appTouchEvent.groovy", "r")
                        for line in event:
                                extractedEvents.write(line)
                        event.close()
+               elif eventMap[i] == "button":
+                       #Write two events subsequently
+                       event = open("eventSimulator/buttonPushedEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+                       extractedEvents.write("\t\t\tbreak\n")
+                       eventCounter = eventCounter + 1
+                       extractedEvents.write("\t\tcase %d:\n" % eventCounter)
+                       event = open("eventSimulator/buttonHeldEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+               elif eventMap[i] == "presence":
+                       #Write two events subsequently
+                       event = open("eventSimulator/presencePresentEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+                       extractedEvents.write("\t\t\tbreak\n")
+                       eventCounter = eventCounter + 1
+                       extractedEvents.write("\t\tcase %d:\n" % eventCounter)
+                       event = open("eventSimulator/presenceLeftEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+               elif eventMap[i] == "doorState":
+                       #Write two events subsequently
+                       event = open("eventSimulator/doorOpenEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+                       extractedEvents.write("\t\t\tbreak\n")
+                       eventCounter = eventCounter + 1
+                       extractedEvents.write("\t\tcase %d:\n" % eventCounter)
+                       event = open("eventSimulator/doorClosedEvent.groovy", "r")
+                       for line in event:
+                               extractedEvents.write(line)
+                       event.close()
+               eventCounter = eventCounter + 1
+
                ###TODO: Add more events later
                extractedEvents.write("\n\t\t\tbreak\n")
        extractedEvents.write("\t}\n")