import os,sys,network,socket,time,machine,gc,tf,btree # these helper classes let us use the tf.transfer() iterator, # by intercepting the .write() class wc(): def __init__(self): self.words=self.lines=self.bytes_=0 def write(self,text): self.bytes_ += len(text) self.lines += 1 self.words += len(text.split()) class lessor(): def __init__(self,nums=False): self.i=0 self.nums=nums def write(self,text): if self.i==-1: return self.i += 1 if self.nums: sys.stdout.write(str(self.i)+' ') sys.stdout.write(text) if self.i%30==0: sys.stdout.write("====> press to see more, N or Q to quit <====\n") g=sys.stdin.read(1) if g=='\n': g=sys.stdin.read(1) if g in "nNqQ": self.i=-1 # the main entry point for the extension # returns True if the command was interpreted def cmd(args): cmd=args[0] if cmd in ('wc','more','less','exec') and len(args)<2: print("syntax: "+cmd+" ") return True if cmd in ('wc','more','less'): nums=False if cmd=='wc': w = wc() else: if args[1]=='-n': nums=True del(args[1]) w = lessor(nums) try: tf.transfer(args[1],w) if cmd=='wc': print("lines: {}\twords: {}\tbytes: {}".format(w.lines, w.words, w.bytes_)) except: print("file not found: "+args[1]) elif cmd in ('ifconfig','ip'): ifc=network.WLAN().ifconfig() print("IP: {}\tmask: {}\tgateway: {}\tDNS: {}".format(*ifc)) elif cmd in ('host','nslookup','dig'): if len(args)<2: print("syntax: host ") else: print("host <{}> is at {}".format(args[1],socket.getaddrinfo(args[1],80)[0][-1][0])) elif cmd=='connect': if len(args)<3: print("syntax: connect ") else: w=network.WLAN(network.STA_IF) w.connect(args[1],args[2]) print("connecting...",end=' ') time.sleep(3) print(w.ifconfig() if w.isconnected() else "not yet connected; try 'ifconfig' in a few seconds") elif cmd=='scan': w=network.WLAN(network.STA_IF) print("scanning...") s=w.scan() if len(s)==0: print("no AP found") return True for i in s: print("ch: {}\tRSSI: {}\t{}\tSSID: {}".format(i[2],i[3],"open" if i[4]==0 else "",i[0])) elif cmd=='freq': # identify esp32 or esp8266 try: # is this esp8266 machine.TouchPad freqs=("160","80","240") except AttributeError: freqs=("160","80") if len(args)==1 or args[1] in freqs: if len(args)>1: machine.freq(int(args[1])*1000000) print("master cpu frequency {}MHz".format(machine.freq()//1000000)) else: print("syntax: freq [ 160 | 80 | 240 ] ") elif cmd=='btree': try: f=open(args[1]) b=btree.open(f) print("Key\t\tValue") i=0 for w in b: print("{}\t{}".format(w,b[w])) i+=1 if i%30==0: r=input("continue? ") if r=='n': break except OSError: print("file not found or is not a btree database") elif cmd=='exec': try: exec(open(args[1]).read(),globals(),globals()) except OSError: print("file not found") elif cmd=='free': print("memory used: {}\tmemory free:{}".format(gc.mem_alloc(),gc.mem_free())) elif cmd=='help': print("==Extended commands") print(" connect \tscan") print(" ifconfig/ip \t\thost/dig/nslookup ") print(" freq [160 | 80 | 240]\texec ") print(" free \t\t\twc ") print(" less/more [-n] ") print(" bt-list ") else: # command not found return False return True