实现自动输入密码

touch.py

@@ -0,0 +1,120 @@
+import os
+import sys
+import time
+import json
+import serial
+
+from ocr import get_text
+from log import log_init, myprint
+
+
+class ser(object):
+  gpio_table={
+     "enter":"pa0",
+     "1":"pa4",
+     "2":"pa5",
+     "3":"pa6",
+     "4":"pb0",
+     "5":"pb1",
+     "6":"pb10",
+     "7":"pa12",
+     "8":"pa11",
+     "9":"pb7",
+     "0":"pb6"
+  }
+  def __init__(self):
+      self.ser=None
+  def open(self,com:str):
+      try:
+          self.ser = serial.Serial(port=com, baudrate=115200, timeout=0.3)
+      except Exception:
+          print(f"serial com:{com} open failed.")
+          sys.exit(-1)
+  def close(self):
+    if not (self.ser is None):
+      self.ser.close()
+      self.ser=None
+  def write(self,t:str):
+    # print("send:",t.encode('utf-8').hex(' '))
+    if not (self.ser is None):
+      self.ser.write(t.encode(encoding="utf-8"))
+  def read(self):
+    data=bytearray()
+    if not (self.ser is None):
+      data=self.ser.read(4096)
+    try:
+      t=data.decode(encoding='utf-8')
+    except Exception as e:
+      print(e)
+      print(data.hex(' '))
+      t=""
+    return t
+  def touch(self,key:str):
+    if not (key in self.gpio_table):
+      return
+    gpio=self.gpio_table[key]
+    print(f"touch {key}:{gpio}")
+    self.write(f"output on {gpio}\r\n")
+    ret=self.read()
+    print(ret)
+    time.sleep(0.05)
+    self.write(f"output off {gpio}\r\n")
+    ret=self.read()
+    print(ret)
+    time.sleep(0.05)
+  def enter(self):
+    s.touch("enter")
+  def passsword(self,p:str):
+    for i in p:
+      self.touch(i)
+
+
+def find_text(t:str):
+  text=get_text(1560,198,363,331)
+  print(text)
+  for item in text:
+    if(item[1]==t):
+      return True
+  return False
+
+
+def data_save(data:dict):
+  data_file="index.json"
+  with open(data_file,mode='w+') as f:
+    f.write(json.dumps(data,sort_keys=True, indent=2, separators=(',', ': ')))
+
+
+def data_get()->dict:
+  data_file="index.json"
+  if os.path.exists(data_file):
+    with open(data_file,mode='r') as f:
+      return json.loads(f.read())
+  return {"index":0}
+
+if __name__ == "__main__":
+  s=ser()
+  s.open("com6")
+  password=data_get().get("index",0)
+  log_init("log.txt")
+  myprint(f"start_index:{password}")
+  while(True):
+    try:
+      if(find_text("手机存储")):
+        myprint("enter")
+        s.enter()
+      if(find_text("请输入解锁密码")):
+        myprint(f"try:{password:06d}")
+        s.passsword(f"{password:06d}")
+    except Exception as e:
+      myprint(e)
+      data_save({"index":password})
+      break
+    except KeyboardInterrupt as k:
+      myprint(k)
+      data_save({"index":password})
+      break
+    if password>999999:
+      myprint("try end")
+      break
+  myprint(f"end_index:{password}")
+  s.close()