python-sht31/main.py at master · zepn/python-sht31 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
"""
 *  NOTE      - main.py
 *  Author    - Aru
 *
 *  Created   - 2022.11.17
 *  Github    - https://github.com/aruyu
 *  Contact   - vine9151@gmail.com
"""

from src import *
from sys import modules, stderr
import time, traceback, sys


has_record_am = False
has_record_pm = False

if __name__ == '__main__':
  """Entry point."""
  debug = False
  try:
    argparser = ArgumentParser(description=modules[__name__].__doc__)
    argparser.add_argument('device', nargs='?', default='ftdi:///?',  help='serial port device name')
    argparser.add_argument('-S', '--no-smb',    action='store_true',  default=False, help='use regular I2C mode vs. SMBbus scan')
    argparser.add_argument('-P', '--vidpid',    action='append',      help='specify a custom VID:PID device ID, may be repeated')
    argparser.add_argument('-V', '--virtual',   type=FileType('r'),   help='use a virtual device, specified as YaML')
    argparser.add_argument('-v', '--verbose',   action='count',       default=0, help='increase verbosity')
    argparser.add_argument('-d', '--debug',     action='store_true',  help='enable debug mode')
    args = argparser.parse_args()
    debug = args.debug

    if not args.device:
      argparser.error('Serial device not specified')

    loglevel = max(DEBUG, ERROR - (10 * args.verbose))
    loglevel = min(ERROR, loglevel)
    if debug:
      formatter = Formatter('%(asctime)s.%(msecs)03d %(name)-20s %(message)s', '%H:%M:%S')
    else:
      formatter = Formatter('%(message)s')
    FtdiLogger.log.addHandler(StreamHandler(stderr))
    FtdiLogger.set_formatter(formatter)
    FtdiLogger.set_level(loglevel)

    if args.virtual:
      from pyftdi.usbtools import UsbTools
      # Force PyUSB to use PyFtdi test framework for USB backends
      UsbTools.BACKENDS = ('pyftdi.tests.backend.usbvirt', )
      # Ensure the virtual backend can be found and is loaded
      backend = UsbTools.find_backend()
      loader = backend.create_loader()()
      loader.load(args.virtual)

    try:
      add_custom_devices(Ftdi, args.vidpid, force_hex=True)
    except ValueError as exc:
      argparser.error(str(exc))

    I2CScan.scan(args.device, not args.no_smb)

  except (ImportError, IOError, NotImplementedError, ValueError) as exc:
    print('\nError: %s' % exc, file=stderr)
    if debug:
        print(format_exc(chain=False), file=stderr)
    exit(1)
  except KeyboardInterrupt:
    exit(2)

  except Exception:
    print(traceback.format_exc())
    print(sys.exc_info()[2])

  else:
    time.sleep(1)
    sht31 = Sht31(args.device)
    save_csv = SaveCsv()
    save_csv.start()

    try:
      while True:
        sht31.start_loop()

        if (has_record_am == False) and (save_csv.get_present_time() == '10'):
          save_csv.make_csv(sht31.get_temperature(), sht31.get_humidity())
          has_record_am = True

        elif (has_record_pm == False) and (save_csv.get_present_time() == '15'):
          save_csv.make_csv(sht31.get_temperature(), sht31.get_humidity())
          has_record_pm = True

        elif save_csv.get_present_time() == '23':
          has_record_am = False
          has_record_pm = False

    except:
      sht31.end_loop()
      save_csv.stop_date_thread()
      print("process cancelled.\n\nreconnect ftdi device once.\n")