feature/driver/timezone: time zones is supported

Author: pandiannano

m5stack/libs/driver/timezone.py

@@ -0,0 +1,83 @@
+from time import localtime
+from socket import socket, getaddrinfo, AF_INET, SOCK_DGRAM
+from struct import unpack
+from errno import ETIMEDOUT
+from machine import RTC
+import network
+
+wlan_sta = network.WLAN(network.STA_IF)
+
+class TZONE(object):
+
+    def __init__(self, zone=0, win=True):
+        self.zone = zone
+        self.win = win
+        # time zones is supported
+        self.TIME_ZONE = {-11: -11, -10: -10, -9: -9, -8: -8, -7: -7, -6: -6, -5: -5,
+        -4: -4, -3: -3, -2: -2, -1: -1, 0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6,
+        7: 7, 8: 8, 9: 9, 10: 10, 11: 11, 12: 12, 13: 13, 14: 14}
+        # months of summer and winter time
+        self.MONTH = {'sum': 3, 'win': 10}  # 3 - march, 10 - october
+
+        self.NTP_DELTA = 2208988800
+        self.host = "cn.pool.ntp.org"
+
+    def getntp(self, host="cn.pool.ntp.org"):
+        # print('Get UTC time from NTP server...')
+        NTP_QUERY = bytearray(48)
+        NTP_QUERY[0] = 0x1b
+        self.host = host
+
+        if wlan_sta.isconnected() is False:
+            raise OSError('Wifi Not Started')
+
+        try:
+            addr = getaddrinfo(self.host, 123)[0][-1]
+        except OSError:  # as exc:
+            #  if exc.args[0] == -2:
+            print('Connect NTP Server: Error resolving pool NTP')
+            return 0
+        s = socket(AF_INET, SOCK_DGRAM)
+        s.settimeout(1)
+        # res = s.sendto(NTP_QUERY, addr)
+        s.sendto(NTP_QUERY, addr)
+
+        try:
+            msg = s.recv(48)
+        except OSError as exc:
+            if exc.args[0] == ETIMEDOUT:
+                print('Connect NTP Server: Request Timeout')
+                s.close()
+                return 0
+        s.close()
+        val = unpack("!I", msg[40:44])[0]
+        return val - self.NTP_DELTA
+
+    def sunday(self, year, month):
+        for d in range(1, 32):
+            a = (14 - month) // 12
+            y = year - a
+            m = month + 12 * a - 2
+            if (((d + y + y // 4 - y // 100 + y // 400 + (31 * m) // 12)) % 7) == 0:
+                if d + 7 > 31:
+                    return d
+
+    def adj_tzone(self, utc):
+        if utc[1] > self.MONTH['sum']:
+            if utc[1] <= self.MONTH['win'] and utc[2] < self.sunday(utc[0], self.MONTH['win']):
+                # print('TIME ZONE Summer:', self.TIME_ZONE[self.zone])
+                return self.TIME_ZONE[self.zone]
+        if utc[1] == self.MONTH['sum'] and utc[2] >= self.sunday(utc[0], self.MONTH['sum']):
+            # print('TIME ZONE Summer:', self.TIME_ZONE[self.zone])
+            return self.TIME_ZONE[self.zone]
+        else:
+            # print('TIME ZONE Winter:', self.TIME_ZONE[self.zone] - 1)
+            return self.TIME_ZONE[self.zone] - 1
+
+    def setzone(self):
+        ntp = self.getntp()
+        z = self.adj_tzone(localtime(ntp)) if self.win else 0
+        utc = localtime(ntp + (3600 * z))
+        # print('Update time for Time Zone: ', z)
+        RTC().datetime(utc)
+        # print('Local Time: ', str(localtime()))