API tools faq
paste
Advertisement
orborbson

nadajnik.py

orborbson
May 5th, 2024 (edited)
25
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 7.60 KB | None | 0 0
raw download clone embed print report
  1. import machine, time, struct, sys, ssd1306
  2. import network, socket, binascii, hashlib, random
  3.  
  4. # ekran
  5. i2c = machine.I2C(1, sda=machine.Pin(21), scl=machine.Pin(22))
  6. ekran = ssd1306.SSD1306_I2C(128, 64, i2c)
  7. ekran.contrast(1)
  8.  
  9. # pzem004t
  10. uart = machine.UART(2, baudrate=9600, bits=8, parity=None, stop=1, tx=16, rx=17)
  11.  
  12. # przycisk dotykowy (TTP223)
  13. przycisk = machine.Pin(27, machine.Pin.IN)
  14.  
  15. ssid = "Master-Slave"
  16. password = "Proste123"
  17. hotspot = False
  18.  
  19. #PRESET = 0xFFFF
  20. #POLYNOMIAL = 0xA001 # bit reverse of 0x8005
  21. def crc16(data, preset=0xFFFF, polynomial=0xA001):
  22.     crc = preset
  23.     for c in data:
  24.         crc = crc ^ c
  25.         for j in range(8):
  26.             if crc & 0x01:
  27.                 crc = (crc >> 1) ^ polynomial
  28.             else:
  29.                 crc = crc >> 1
  30.     return data + crc.to_bytes(2, "little") # ESP32 - little-endian
  31.  
  32. def read_all():
  33.     if uart.write(req_crc):
  34.         time.sleep_ms(100)
  35.         payload = uart.read()
  36.         if payload == None:
  37.             raise ValueError("PZEM004T")
  38.         payload = payload[3:-2]
  39.         fmt = ">" + (int(len(payload)/2)) * "H"
  40.         return struct.unpack(fmt, payload)
  41.    
  42. def resetuj_licznik_energii():
  43.     uart.write(b'\xF8\x42\xC2\x41')
  44.    
  45. def pokaz_mac():
  46.     ekran.fill(0)
  47.     #ekran.contrast(1)
  48.     ekran.text("= Master-Slave =".center(16), 0, 1)
  49.     ekran.text("v0.01".center(16), 0, 12)
  50.     ekran.text("Adres MAC:".center(16), 0, 40)
  51.     ekran.text(binascii.hexlify(network.WLAN().config("mac")).decode().upper().center(16), 0, 55)
  52.     ekran.show()
  53.    
  54. def przycisk_konfig(self):
  55.     global hotspot
  56.     hotspot = True
  57.     ekran.fill(0)
  58.     #ekran.contrast(1)
  59.     y = 1
  60.     ekran.text("= Konfiguracja =".center(16), 0, y); y += 16
  61.     ekran.text("HOTSPOT:", 0, y); y += 12
  62.     ekran.text(ssid, 2, y); y += 16
  63.     ekran.text("HASLO:", 0, y); y += 12
  64.     ekran.text(password, 2, y)
  65.     ekran.show()
  66.    
  67. def oled_wyswietl(txt):
  68.     ekran.fill(0)
  69.     #ekran.contrast(1)
  70.     licz = 1
  71.     for i in txt:
  72.         ekran.text(i, 1, licz)
  73.         licz += 11
  74.     ekran.show()
  75.    
  76. def oled_error(s):
  77.     ekran.fill(0)
  78.     ekran.text("[BLAD]:", 0, 0)
  79.     ekran.text(s, 0, 16)
  80.     ekran.show()
  81.    
  82. def generuj_hash():
  83.     hash = hashlib.md5("%.20f" % random.random())
  84.     return str(binascii.hexlify(hash.digest()), "utf-8")
  85.    
  86. def hotspotx(ssid, password):
  87.     ap = network.WLAN(network.AP_IF)
  88.     ap.active(True)
  89.     ap.config(essid=ssid, password=password, authmode=network.AUTH_WPA_WPA2_PSK)
  90.  
  91.     while ap.active() == False:
  92.       pass
  93.  
  94.     print("AP działa:", str((ap.ifconfig())[0]))
  95.    
  96.     hash = generuj_hash()
  97.     http_ok = """<html><head>
  98.                <meta charset="UTF-8", name="viewport" content="width=device-width, initial-scale=1">
  99.                <link rel="shortcut icon" href="#" />
  100.                </head>
  101.                <body><h2>%s</h2><hr>
  102.                <p><a href="resetuj_esp-%s">Resetuj ESP</a></p>
  103.                <p><a href="resetuj_licznik-%s">Resetuj licznik energii</a></p>
  104.                <p><a href="wyjdz-%s">Wyjdź</a></p>
  105.                </body></html>"""% (ssid, hash, hash, hash)
  106.  
  107.     http_404 = """<html><head>
  108.                    <meta charset="UTF-8", name="viewport" content="width=device-width, initial-scale=1">
  109.                    <link rel="shortcut icon" href="#" />
  110.                    <meta http-equiv="refresh" content="2; url=/">
  111.                    </head>
  112.                    <body><h2>[404] Nie znaleziono!</h2></body></html>"""
  113.    
  114.     http_licznik = """<html><head>
  115.                    <meta charset="UTF-8", name="viewport" content="width=device-width, initial-scale=1">
  116.                    <link rel="shortcut icon" href="#" />
  117.                    <meta http-equiv="refresh" content="2; url=/">
  118.                    </head>
  119.                    <body><h2>Zresetowano!</h2></body></html>"""
  120.  
  121.  
  122.     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  123.     s.bind(("", 80))
  124.     s.listen(5)
  125.    
  126.     while True:
  127.         conn, addr = s.accept()
  128.         print("Otrzymałem połączenie z %s" % str(addr))
  129.         request = conn.recv(1024)
  130.         req = (str(request, "utf-8").splitlines()[0]).split(" ")[1]
  131.         print("Otrzymano zapytanie:", req)
  132.        
  133.         html_data = ""
  134.        
  135.         if req == "/" or req == "/index.html" or req == "/favicon.ico":
  136.             print("=====> HTML <======")
  137.             html_data = http_ok
  138.         elif req == ("/resetuj_esp-" + hash):
  139.             print("=====> reset esp <======")
  140.             conn.send(html_data)
  141.             conn.close()
  142.             machine.reset()
  143.         elif req == ("/resetuj_licznik-" + hash):
  144.             print("=====> reset licznika <======")
  145.             html_data = http_licznik
  146.             resetuj_licznik_energii()
  147.         elif req == ("/wyjdz-" + hash):
  148.             print("=====> wyjdź <======")
  149.             html_data = http_ok
  150.             conn.send(html_data)
  151.             conn.close()
  152.             time.sleep(5)
  153.             machine.reset()
  154.         else:
  155.             html_data = http_404
  156.         conn.send(html_data)
  157.         conn.close()
  158.  
  159. pokaz_mac()
  160. przycisk.irq(trigger=machine.Pin.IRQ_RISING, handler=przycisk_konfig)
  161. time.sleep(5)
  162.  
  163.  
  164. if hotspot:
  165.     hotspotx(ssid, password)
  166. else:
  167.     #req_crc = crc16(b"\xF8\x04\x00\x03\x00\x02")
  168.     req_crc = crc16(b"\xF8\x04\x00\x00\x00\x0A")
  169.  
  170.  
  171.     def fun(self):
  172.         ekran.poweroff()
  173.  
  174.     tim = machine.Timer(0)
  175.     timEkranOff = machine.Timer(1)
  176.     ekran_wl = False
  177.  
  178.     #przycisk_licznik = 0
  179.     def przycisk_fun(self):
  180.         ekran.poweron()
  181.         tim.init(period = 5000, mode=machine.Timer.ONE_SHOT, callback=fun)
  182.  
  183.     przycisk.irq(trigger=machine.Pin.IRQ_RISING, handler=przycisk_fun)
  184.  
  185.     ekran_wl = True
  186.     while True:
  187.         try:
  188.             data = read_all()
  189.         except KeyboardInterrupt:
  190.             print("Przerwano: Ctrl + C")
  191.             break
  192.         except Exception as e:
  193.             print("[BŁĄD]:", str(e))
  194.             oled_error(str(e))
  195.             break
  196.         else:
  197.             if data:
  198.                 volt = data[0]/10.0
  199.                 curr = ((data[2]<<16) |  (data[1]))/1000.0
  200.                 power = ((data[4]<<16) |  (data[3]))/10.0
  201.                 energy = ((data[6]<<16) |  (data[5]))/1000.0
  202.                 freq = data[7]/10.0
  203.                 pf = data[8]/100.0
  204.                 txt = ( "{0:<5}{1:.1f}".format("V:", volt),
  205.                         "{0:<5}{1:.3f}".format("A:", curr),
  206.                         "{0:<5}{1:.1f}".format("W:", power),
  207.                         "{0:<5}{1:.3f}".format("kWh:", energy),
  208.                         "{0:<5}{1:.2f}".format("PF:", pf),
  209.                         "{0:<5}{1:.1f}".format("Hz:", freq)
  210.                         )
  211.                 oled_wyswietl(txt)
  212.                
  213.                 if power:
  214.                     if not ekran_wl:
  215.                         print("Ekran ON")
  216.                         ekran.poweron()
  217.                         ekran_wl = True
  218.                 elif ekran_wl:
  219.                     print("Ekran OFF")
  220.                     ekran_wl = False
  221.                     tim.init(period = 5000, mode=machine.Timer.ONE_SHOT, callback=fun)
  222.                 """
  223.                if power == 0 and wyl == False:
  224.                    wyl = True
  225.                    tim.init(period = 10000, mode=machine.Timer.ONE_SHOT, callback=fun)
  226.                elif power > 0 and wyl == True:
  227.                    wyl = False
  228.                    ekran.poweron()
  229.                """
  230.                
  231.         finally:
  232.             time.sleep_ms(800)
  233.  
  234.     if not ekran_wl:
  235.         ekran.poweron()
  236.     time.sleep(10)
  237.     machine.reset()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!