#!/usr/bin/env python3
import snapshot
import os
import time
import logging

if __name__ == "__main__":
    png = "/home/alientek/source/paddle_ocr/screen_shot.png"
    try:

        with open(f'{png}.log', 'a') as f:
            f.write("start paddle_ocr")
        if os.path.exists(f"{png}"):
            os.remove(f"{png}")
        if os.path.exists(f"{png}.h"):
            os.remove(f"{png}.h")
        if os.path.exists(f"{png}.log"):
            os.remove(f"{png}.log")

        time.sleep(1)
        snapshot.take_screenshot(png)
        snapshot.screenshot_to_text(png)
    except Exception as e:
        logging.exception(e)
        with open(f'{png}.log', 'a') as f:
            f.write(f"exception: {e}")