AdvantageScout/ImageEncodeFiles.py at main · Mechanical-Advantage/AdvantageScout · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import base64
import binascii
import os
from PIL import Image
import io
import psycopg2
import sqlite3 as sql
db_global = "global.db"
db_games = "data_$GAME.db"
conn_global = sql.connect(db_global)
cur_global = conn_global.cursor()
event = cur_global.execute("SELECT value FROM config WHERE key='event'").fetchall()[0][0]
cur_global.execute("SELECT value FROM config WHERE key = 'game'")
game = str(cur_global.fetchall()[0][0])
conn_global.close()
# conn_game = sql.connect(db_games.replace("$GAME", game))
# cur_game = conn_game.cursor()
# TeamImage = cur_game.execute("SELECT Team, Image FROM pit WHERE Event=?", (event,)).fetchall()
conn_grafana = psycopg2.connect(database="Grafana-Output",
                        host="127.0.0.1",
                        user="postgres",
                        password="MA6328",
                        port="5432")
cur_grafana = conn_grafana.cursor()
print(event)
path = "c://mascout//advantagescout//images//"
dir_list = os.listdir(path)
print("Files and directories in '", path, "' :")
eventImages = []
for img_name in range(len(dir_list)):
    if event in dir_list[img_name]:
        print(event in (dir_list[img_name]))
        eventImages.append(dir_list[img_name])
print(eventImages)
# cur_grafana.execute("DELETE FROM Images WHERE Event=?", (event,))
# sql_text = 'DELETE FROM "Images" WHERE "Event"=%s;'
# sql_data = (event,)
# cur_grafana.execute(sql_text, sql_data)
# conn_grafana.commit()
count=-1
data=[" "," "," "]
eventImages.sort()
firstImage = eventImages[0].split("-")
currentTeam = firstImage[1]


for row in eventImages:
    SplitRow = row.split("-")
    img_name = row[1]
    img_name = "images//" + row
    if currentTeam != SplitRow[1]:
        count = 0
        sql_text = 'DELETE FROM "Images" WHERE "Event"=%s And "Team" = %s;'
        sql_data = (event, currentTeam,)
        cur_grafana.execute(sql_text, sql_data)
        conn_grafana.commit()
        sql_text = 'INSERT INTO "Images" ("Event", "Team", "Image", "Image2","Image3") VALUES (%s,%s,%s,%s,%s)'
        sql_data = (event, currentTeam, data[0],data[1],data[2],)
        currentTeam = SplitRow[1]
        data=[" "," "," "]
    # cur_grafana.execute("INSERT INTO Images (Event, Team, Image) VALUES (?, ?, ?)",
    #                     (event, row[0], data))
        cur_grafana.execute(sql_text, sql_data)
        conn_grafana.commit()
    else:
        count = count + 1

    if os.path.isfile(img_name):

        with open(img_name, "rb") as image_file:
            tempData = base64.b64encode(image_file.read()).decode('utf-8')
            print ("Processing image")
            print(count)
            print(currentTeam)
            data[count] = tempData
sql_text = 'DELETE FROM "Images" WHERE "Event"=%s And "Team" = %s;'
sql_data = (event, currentTeam,)
cur_grafana.execute(sql_text, sql_data)
conn_grafana.commit()
sql_text = 'INSERT INTO "Images" ("Event", "Team", "Image", "Image2","Image3") VALUES (%s,%s,%s,%s,%s)'
sql_data = (event, currentTeam, data[0],data[1],data[2],)
currentTeam = SplitRow[1]
data=[" "," "," "]
    # cur_grafana.execute("INSERT INTO Images (Event, Team, Image) VALUES (?, ?, ?)",
    #                     (event, row[0], data))
cur_grafana.execute(sql_text, sql_data)
conn_grafana.commit()