#!/usr/bin/python3
# -*- coding:utf-8 -*-
import sys
import time
import logging
from PIL import Image

LIB_DIR = '/home/ubuntu/workplace/e-Paper/RaspberryPi_JetsonNano/python/lib'
sys.path.append(LIB_DIR)

from waveshare_epd import epd2in13_V4

logging.basicConfig(level=logging.INFO)


def main():
    epd = epd2in13_V4.EPD()
    epd.init()

    image = Image.open(sys.argv[1]).convert('1')
    image = image.rotate(180)

    epd.display(epd.getbuffer(image))
    time.sleep(2)
    epd.sleep()


if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        epd2in13_V4.epdconfig.module_exit(cleanup=True)
        raise