Auto upload

Author: Daniel-Chin

console.py

@@ -1,15 +1,18 @@
 '''
-Windows only. If not windows, start IPython. 
-Advantage over IPython: 
+Windows only. If not windows, console() calls IPython.embed(). 
+Advantages over IPython: 
     1. Lighter
     2. Other threads can still print things when user is inputting commands
-Issue: 
-    If you wanna scroll up, you need to input().
-    Changing global bindings doesn't work. Sorry. 
+    3. Tab auto-completes your phrase, even under Windows! (I'm proud.) 
+Issues: 
+    1. If you wanna scroll up, you need to input().
+    2. Reassigning module global variables will not be visible to module native codes. Sorry. 
 '''
 import platform
-from listen import listen
+from interactive import listen, strCommonStart
 from kernal import Kernal
+from graphic_terminal import clearLine
+import string
 
 CURSOR = '|'
 
@@ -18,7 +21,7 @@ def console(namespace = {}, prompt = '>>> ', use_input = False, fixer = None):
         from IPython import embed
         embed()
         return
-    print('console.console Warning: no support for change in global bindings. ')
+    print('console.console Warning: no support for reassigning module global variables. ')
     history = []
     kernal = Kernal(namespace)
     next(kernal)
@@ -38,7 +41,8 @@ def console(namespace = {}, prompt = '>>> ', use_input = False, fixer = None):
                 op = listen(timeout = .5)
                 last_len = len(command)
                 if op == b'\r':
-                    print(prompt + command.replace('\x1a', '^Z') + ' ')
+                    clearLine()
+                    print(prompt + command.replace('\x1a', '^Z'))
                     break
                 elif op is None:
                     cursor_bright = not cursor_bright
@@ -100,18 +104,53 @@ def console(namespace = {}, prompt = '>>> ', use_input = False, fixer = None):
                                     word_ended = True
                         else:
                             cursor = len(command)
+                    elif op == b'\t':
+                        # auto complete
+                        legal_prefix = string.ascii_letters + string.digits + '_'
+                        reversed_names = []
+                        name_end = cursor
+                        name_start = cursor - 1
+                        while True:
+                            if name_start >= 0 and command[name_start] in legal_prefix:
+                                name_start -= 1
+                            else:
+                                reversed_names.append(command[name_start + 1:name_end])
+                                name_end = name_start
+                                name_start -= 1
+                                if name_start < 0 or command[name_end] != '.':
+                                    break
+                        keyword = reversed_names.pop(0)
+                        names = reversed(reversed_names)
+                        to_search = kernal.send('dir(%s)' % '.'.join(names))
+                        if len(reversed_names) == 0:
+                            # include builtins
+                            to_search += dir(__builtins__)
+                        next(kernal)
+                        candidates = [x for x in to_search if x.startswith(keyword)]
+                        if len(candidates) >= 1:
+                            if len(candidates) == 1:
+                                to_become = candidates[0]
+                            if len(candidates) > 1:
+                                clearLine()
+                                print('auto-complete: ', end = '')
+                                [print(x, end = '\t') for x in candidates]
+                                print()
+                                to_become = strCommonStart(candidates, len(keyword))
+                            to_insert = to_become[len(keyword):]
+                            command = command[:cursor] + to_insert + command[cursor:]
+                            cursor += len(to_insert)
                     elif op[0] in range(1, 26) or op[0] in (0, 224):
                         pass
                     else:
                         command = command[:cursor] + op.decode() + command[cursor:]
                         cursor += 1
-                padding = max(0, last_len - len(command)) * 2
                 if cursor_bright:
                     cursor_show = CURSOR
                 else:
                     cursor_show = '_'
                 cursed_command = command[:cursor] + cursor_show + command[cursor:]
-                print(prompt + cursed_command.replace('\x1a', '^Z') + ' '*padding, end='\r')
+                clearLine()
+                print(prompt + cursed_command.replace('\x1a', '^Z'), end='\r')
         if fixer is not None:
             command = fixer(command)
         if command in ('exit', 'exit()', '\x1a'):
@@ -123,7 +162,10 @@ def console(namespace = {}, prompt = '>>> ', use_input = False, fixer = None):
         if command == '':
             continue
         history.append(command)
-        kernal.send(command)
+        result = kernal.send(command)
+        next(kernal)
+        if result is not None:
+            print(result)
 
 if __name__ == '__main__':
     console({})

fakep2p.py

@@ -0,0 +1,9 @@
+from terminalsize import get_terminal_size
+
+def clearLine():
+    '''
+    Clears the current line, then \r. 
+    Used when previously printed without \n. 
+    '''
+    terminal_width = get_terminal_size()[0] - 4
+    print('\r', ' ' * terminal_width, end = '\r', sep = '')

graphic_terminal.py

@@ -0,0 +1,97 @@
+'''
+Terminal interactivity utils. 
+'''
+#===============================================================================
+# I do not know why this was here. 
+# if 'flush' not in getargspec(print).args:
+#     print_3 = print
+#     def print(*args, flush = False, **kw):
+#         print_3(*args, **kw)
+#===============================================================================
+__all__ = ['listen', 'strCommonStart']
+from time import sleep
+FPS = 30
+
+try:
+    import msvcrt
+    import sys
+    def getFullCh():
+        first=msvcrt.getch()
+        if first in (b'\x00', b'\xe0'):
+            return first + msvcrt.getch()
+        else:
+            return first
+    # if sys.getwindowsversion() >= (10, 0, 17134) and False: # Strange windows update on 2018/10/22
+    #     __getFullCh = getFullCh
+    #     def getFullCh():
+    #         ch = __getFullCh()
+    #         if len(ch) == 1:
+    #             assert msvcrt.getch() == b'\x00'
+    #         return ch
+except ImportError:
+    msvcrt = None
+
+def listen(choice=None, timeout=0):
+    '''
+    choice can be a list of choices or a single choice. 
+    Elements can be b'' or ''.
+    If timeout=0, it's blocking. 
+    timeout is in second. 
+    Supports non-windows. 
+    '''
+    if choice is not None:
+        if type(choice) in (bytes, str):
+            choice = (choice, )
+        bChoice=[]
+        for i in choice:
+            if type(i) is bytes:
+                bChoice.append(i)
+            else:
+                bChoice.append(i.encode())
+    print('', end = '', flush = True)     # Just to flush
+    if msvcrt is None:
+        if choice is None:
+            op = eval('b"%s"' % input())
+            if op == b'':
+                return b'\r'    # So android doesn't need to type "\r"
+        else:
+            print(bChoice)
+            op = None
+            while op not in bChoice:
+                op = eval('b"%s"' % input())
+                if op == b'' and b'\r' in bChoice:
+                    return b'\r'    # So android doesn't need to type "\r"
+            return op
+    if timeout != 0:
+        for i in range(int(timeout*FPS)):
+            if msvcrt.kbhit():
+                op = getFullCh()
+                if choice is None or op in bChoice:
+                    return op
+            sleep(1/FPS)
+        return None
+    op=getFullCh()
+    while not (choice is None or op in bChoice):
+        op=getFullCh()
+    return op
+
+def strCommonStart(list_strs, known_len = 0):
+    '''
+    Find the common start for a list of strings. 
+    Useful for auto-complete for the user. 
+    `known_len` is known length of common start - for performance. 
+    '''
+    columns = zip(* list_strs)
+    i = -1  # in case one string is ''
+    for i, column in enumerate(columns):
+        if i >= known_len:
+            shifted_column = iter(column)
+            next(shifted_column)
+            if not all(x == y for x, y in zip(column, shifted_column)):
+                i -= 1
+                break
+    return list_strs[0][:i + 1]
+
+if __name__=='__main__':
+    from console import console
+    console(globals())

interactive.py

@@ -19,22 +19,20 @@ def cls():
             if __command == 'exit':
                 return
             try:
-                __result = eval(__command + '\r')
-                if __result is not None:
-                    print(__result)
-                del __result
-                __eval_good = True
+                yield eval([__command + '\r', exec('del __command')][0])
+                # Tries to hide `__command` in dir(). 
+                # doesn't work tho. 
+                # But you have to admit, it's a clever try. 
+                continue
             except SyntaxError:
-                __eval_good = False
-            if not __eval_good:
-                del __eval_good
-                exec(__command + '\r')
-            else:
-                del __eval_good
+                pass
+            exec(__command + '\r')
+            yield None
         except:
             import traceback
             traceback.print_exc()
             del traceback
+            yield None
 
 if __name__ == '__main__':
     kernal = Kernal({})