import psycopg2
import time
from datetime import datetime, timedelta
from database.database import DB

   
class pothole():
    def __init__(self):
        start_time=time.time()
        self.db=DB()

    def report(self,report_id,pothole_id,pothole_x,pothole_y,z):
        self.db = DB()
        self.db.db_add_report(report_id,pothole_x,pothole_y,z)
        result=self.db.db_get_near_point(pothole_x,pothole_y)
        if len(result) >3 :
            value = 0
            for i in range(len(result)):
                value += result[i][0]
            average_x= (value / len(result))
            value2 = 0
            for i in range(len(result)):
                value2 += result[i][1]
            average_y= (value2 / len(result))
            self.db.db_delete_pothole(average_x,average_y)
            self.db.db_add_pothole(pothole_id,average_x,average_y)

    def display(self,timestamp):
        self.db = DB()
        value=self.db.db_display_pothole(timestamp)
        return value