auto_screenshot.py

# -*- coding: utf-8 -*-
"""
Created on Sun Jun 18 21:34:22 2017

@author: qzane
"""
from PIL import Image
import pyautogui as auto
from time import sleep
import os
import sys

# a = auto.screenshot()
# auto.press('up,left,right,down')
# auto.scroll(4)


def read_next_page():
    auto.press('right')
    auto.click()
    sleep(0.6)
    auto.press('down',3)
    sleep(0.2)
    a = auto.screenshot()
    auto.press('down',51)
    sleep(0.2)
    b = auto.screenshot()
    return (a,b)

def read_next_pages():
    auto.press('right')
    auto.click()
    sleep(1)
    a = auto.screenshot()
    auto.press('down',51)
    sleep(0.2)
    b = auto.screenshot()
    return (a,b)

def crop(a,b):
    #a = a.crop((255,0,1715,1050))
    #b = b.crop((255,25,1715,1080))
    a = a.crop((255,0,1715,1050)).resize((1217, 875), Image.ANTIALIAS)
    b = b.crop((255,25,1715,1080)).resize((1217, 879), Image.ANTIALIAS)
    target = Image.new('RGB',(1240,1754),(255,255,255))
    target.paste(a, (0,0))
    target.paste(b, (0, 870))
    return target

def main2():
    if not os.path.isdir('down_good1'):
        os.mkdir('down_good1')
    for i in range(0,187):
        a = Image.open(os.path.join('gui_down','p%d.1.bmp'%i))
        b = Image.open(os.path.join('gui_down','p%d.2.bmp'%i))
        target = crop(a,b)
        target.save(os.path.join('down_good1','p%d.bmp'%(i+2)))#, quality=99)
        sys.stderr.write("%s\n"%i)
        

def main():
    if not os.path.isdir('gui_down'):
        os.mkdir('gui_down')
    sleep(10)
    for i in range(200):
        a,b = read_next_pages()
        a.save(os.path.join('gui_down','p%s.1.bmp'%i))#, quality=95)
        b.save(os.path.join('gui_down','p%s.2.bmp'%i))#, quality=95)