air-pollution/rds.py at main · raspberry-cookie/air-pollution · 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
import pymysql
from database import check_rds_tables
from database import insert_rds
from database import update_rds

def rds_device():
    db = pymysql.connect(host='localhost',user= 'root',password='raspberry',db='raspdb',charset='utf8')
    cur = db.cursor()

    sql = "SELECT * FROM device"
    cur.execute(sql)
    row = cur.fetchall()

    for r in row:
        # 1이면 중복id  존재(기본키는 중복될 수 없다함)
        if (check_rds_tables.check_device(r[0]) == 1):
            # id가 이미 내부DB 테이블에 있으면 그 데이터를 update
            update_rds.update_rds_device(r[0],r[1],r[2])
        else:
            # id가 테이블에 없으면 insert
            insert_rds.insert_rds_device(r[0],r[1],r[2])

def rds_location():
    db = pymysql.connect(host='localhost',user= 'root',password='raspberry',db='raspdb',charset='utf8')
    cur = db.cursor()

    sql = "SELECT * FROM location"
    cur.execute(sql)
    row = cur.fetchall()

    for r in row:
        print(r)
        # 1이면 중복id  존재(기본키는 중복될 수 없다함)
        if (check_rds_tables.check_location(r[0]) == 1):
            # id가 이미 내부DB 테이블에 있으면 그 데이터를 update
            update_rds.update_rds_location(r[0],r[1],r[2], r[3])
        else:
            # id가 테이블에 없으면 insert
            insert_rds.insert_rds_location(r[0],r[1],r[2], r[3])

def rds_air_quality_sensor():
    db = pymysql.connect(host='localhost',user= 'root',password='raspberry',db='raspdb',charset='utf8')
    cur = db.cursor()

    sql = "SELECT geohash,device_id, AVG(ozone), AVG(co), AVG(pm10), AVG(pm25) FROM air_quality_sensor GROUP BY (geohash)"
    cur.execute(sql)
    row = cur.fetchall()
    for r in row:
        # 1이면 중복 geohash  존재(기본키는 중복될 수 없다함)
        if (check_rds_tables.check_air_quality(r[0]) == 1):
            # geohash가 이미 내부DB 테이블에 있으면 그 데이터를 update
            update_rds.update_rds_air_quality(r[0],r[1], r[2], r[3],r[4],r[5])
        else:
            # geohash가 테이블에 없으면 insert
            insert_rds.insert_rds_air_quality(r[0],r[1], r[2], r[3],r[4],r[5])

if __name__ == "__main__":
    rds_device()
    rds_location()
    rds_air_quality_sensor()