Added Excerpt2Code and excerpt()

dtmilano · dtmilano · commit 192131bbb2cb · 2013-06-08T02:32:35.000-04:00
- Reformat
- culebra: Added interactive support
diff --git a/AndroidViewClient/src/com/dtmilano/android/viewclient.py b/AndroidViewClient/src/com/dtmilano/android/viewclient.py
@@ -919,15 +919,15 @@ def EndElement(self, name):
         if name == 'hierarchy':
             pass
         elif name == 'node':
-            self.nodeStack.pop( )
+            self.nodeStack.pop()
 
     def CharacterData(self, data):
         '''
         Expat character data event handler
         '''
         
-        if data.strip( ):
-            data = data.encode( )
+        if data.strip():
+            data = data.encode()
             element = self.nodeStack[-1]
             element.cdata += data
 
@@ -942,6 +942,52 @@ def Parse(self, uiautomatorxml):
         parserStatus = parser.Parse(uiautomatorxml, 1)
         return self.root
 
+class Excerpt2Code():
+    ''' Excerpt XML to code '''
+
+    def __init__(self):
+        self.data = None
+
+    def StartElement(self, name, attributes):
+        '''
+        Expat start element event handler
+        '''
+        if name == 'excerpt':
+            pass
+        else:
+            warnings.warn("Unexpected element: '%s'" % name)
+
+    def EndElement(self, name):
+        '''
+        Expat end element event handler
+        '''
+
+        if name == 'excerpt':
+            pass
+
+    def CharacterData(self, data):
+        '''
+        Expat character data event handler
+        '''
+
+        if data.strip():
+            data = data.encode()
+            if not self.data:
+                self.data = data
+            else:
+                self.data += data
+
+    def Parse(self, excerpt):
+        # Create an Expat parser
+        parser = xml.parsers.expat.ParserCreate()
+        # Set the Expat event handlers to our methods
+        parser.StartElementHandler = self.StartElement
+        parser.EndElementHandler = self.EndElement
+        parser.CharacterDataHandler = self.CharacterData
+        # Parse the XML
+        parserStatus = parser.Parse(excerpt, 1)
+        return self.data
+
 class ViewClient:
     '''
     ViewClient is a I{ViewServer} client.
@@ -2053,6 +2099,14 @@ def isKeyboardShown(self):
             return "mInputShown=true" in dim
         return False
 
+    @staticmethod
+    def excerpt(str, execute=False):
+        code = Excerpt2Code().Parse(str)
+        if execute:
+            exec code
+        else:
+            return code
+
 
 if __name__ == "__main__":
     try:
diff --git a/AndroidViewClient/tools/culebra b/AndroidViewClient/tools/culebra
@@ -47,7 +47,7 @@ try:
 except:
     pass
 
-from com.dtmilano.android.viewclient import ViewClient
+from com.dtmilano.android.viewclient import ViewClient, View
 
 from com.android.monkeyrunner import MonkeyRunner, MonkeyDevice
 
@@ -464,32 +464,23 @@ vc.dump(window=options[WINDOW])
 vc.traverse(transform=transform)
 
 if options[INTERACTIVE]:
-    sys.path.insert(0, '/Users/diego/Work/workspace/AndroidEventRecorder/src')
-    print sys.path
-    import com.dtmilano.android.eventrecorder
-    #from com.dtmilano.android.eventrecorder import EventRecorder
-#     try:
-#         #cmdStr = "/bin/ls -l"
-#         cmdStr = "/Users/diego/Work/workspace/AndroidEventPlayer/samples/runme.sh"
-#         #cmdStr = '%(ANDROID_EVENT_PLAYER_HOME)s/src/com/dtmilano/android/eventrecorder.py -v -n -d %(TIMED)s %(DEVICE)s' % {'ANDROID_EVENT_PLAYER_HOME': '/Users/diego/Work/workspace/AndroidEventPlayer', 'TIMED': '', 'DEVICE': '/dev/input/event0'}
-#         print cmdStr
-#         cmd = cmdStr.split()
-#         
-#         pipe = subprocess.Popen(cmd, shell=False) #, stdout=subprocess.PIPE)
-#         if not pipe:
-#             print >>sys.stderr, "Cannot create pipe"
-#             sys.exit(2)
-#         print "pipe", pipe
-#         pipe.communicate()
-#         
-#         while True:
-#             print >>sys.stderr, "reading..."
-#             line = pipe.stdout.readline()
-#             if not line:
-#                 break
-#             print line
-#             self.pipe.poll()
-#             if self.pipe.returncode is not None:
-#                 break
-#     except Exception, e:
-#         print >>sys.stderr, "ERROR:", e
+    import socket
+    HOST = 'localhost'
+    PORT = 8900
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.connect((HOST, PORT))
+    s.sendall("RECORD EVENTS START\n")
+    fin = open("/dev/tty")
+    while True:
+        print >>sys.stderr, "Reading events..."
+        data = s.recv(1024)
+        code = ViewClient.excerpt(data)
+        exec code
+        # unfortunately, this does not work with monkeyrunner
+        #resp = raw_input("Continue recording events? [Y/n]: ")
+        print >>sys.stderr, "Continue recording events? [Y/n]: "
+        resp = fin.readline().strip()
+        if resp in ['N', 'n']:
+            break
+    s.sendall("RECORD EVENTS STOP\n")
+    s.close()
