AdvantageScout/ImageEncodeFilesExternal.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import base64
import binascii
import os
from PIL import Image
import io
import psycopg2
import sqlite3 as sql
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-n', '--newonly', action='store_true')
args = parser.parse_args()
newonly = args
print("args ", args)
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="5.tcp.ngrok.io",
                        user="postgres",
                        password="MA6328",
                        port="23010")
cur_grafana = conn_grafana.cursor()
sql_text = 'SELECT "Team" FROM "Images" where "Event"=%s'
sql_data = (event, )
cur_grafana.execute(sql_text,sql_data)
processed = cur_grafana.fetchall()
processed = [str(x[0]) for x in processed]
print(type(processed[0]))
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]
print (processed)

for row in eventImages:
    SplitRow = row.split("-")
    img_name = row[1]
    img_name = "images//" + row
    print ("Current team "+ SplitRow[1] + " - " +currentTeam)
    if currentTeam not in processed or newonly == False:
        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 " + str(count + 1) + " team " + currentTeam)
                if count < 3:
                    data[count] = tempData
    else:
        currentTeam = SplitRow[1]
if currentTeam not in processed or newonly == False:
    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()