From c39f1fd0c0f86f69df65327950d476e2bd0de390 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurentfainsin@protonmail.com>
Date: Sun, 20 Mar 2022 16:54:58 +0100
Subject: [PATCH] feat: poc database script

---
 .gitignore |  1 +
 src/db.py  | 16 ++++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/.gitignore b/.gitignore
index f952d5c..5fd554b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 .venv
 .mypy_cache
+data
diff --git a/src/db.py b/src/db.py
index 5e390e3..a3dca12 100644
--- a/src/db.py
+++ b/src/db.py
@@ -1,5 +1,21 @@
 import sqlite3
+import uuid
+from fileinput import filename
 
 DATA_FOLDER = "data/"
 
 con = sqlite3.connect("dataset.db")
+
+from ppadb.client import Client as AdbClient
+
+client = AdbClient(host="127.0.0.1", port=5037)
+device = client.device("13231FDD4003UE")
+
+filename = uuid.uuid1()
+result = device.screencap()
+with open(f"{DATA_FOLDER}/{filename}.png", "wb") as fp:
+    fp.write(result)
+    con.execute(f"INSERT INTO data VALUES ('{filename}',{0},{0},{0.5},{0.5},{0},{0},{0},{0},{0})")
+    con.commit()
+
+con.close()