ENSEEIHT/projet-reseaux-profond
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge branch 'master' of git.inpt.fr:tocard-inc/enseeiht/projet-classification

Browse source
This commit is contained in:
gdamms 2022-03-23 18:37:39 +01:00
parent 4dac904664 79aaa6e4f4
commit 9cc60bdec3
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/database.py
View file

@ -1,4 +1,5 @@
import io import io
import logging
import sqlite3 import sqlite3
import uuid import uuid
@ -9,13 +10,16 @@ DATA_FOLDER = "data/"
def take_screenshot(device): def take_screenshot(device):
result = device.screencap() result = device.screencap()
logging.debug("screenshot taken")
image = Image.open(io.BytesIO(result)) image = Image.open(io.BytesIO(result))
image = image.crop((0, 160, 2200, 860)) image = image.crop((0, 160, 2200, 860))
image.thumbnail((150, 150)) image.thumbnail((150, 150))
logging.debug("screenshot converted")
# image = image.resize((150, 150), Image.ANTIALIAS) # image = image.resize((150, 150), Image.ANTIALIAS)
filename = uuid.uuid1() filename = uuid.uuid1()
image.save(f"{DATA_FOLDER}/{filename}.png") image.save(f"{DATA_FOLDER}/{filename}.png")
logging.debug("screenshot saved")
return filename return filename