#!/usr/bin/python3
# Very simple control script for Korad KD3005P Lab Power Supply

import argparse
import serial
import time

commands = ('ident', 'on', 'off', 'off-on')

parser = argparse.ArgumentParser()
parser.add_argument("port", help="serial device, e.g. /dev/ttyACM0")
parser.add_argument("command", choices=commands, help="command to execute")
args = parser.parse_args()

with serial.Serial(args.port, timeout=1) as port:
    if args.command == 'ident':
        port.write(b'*IDN?')
        print(port.read(30).decode('ascii'))
    elif args.command == 'on':
        port.write(b'OUT1')
    elif args.command == 'off':
        port.write(b'OUT0')
    elif args.command == 'off-on':
        port.write(b'OUT0')
        time.sleep(1)
        port.write(b'OUT1')
