from google.cloud import storage
import time
import firebase_admin
from firebase_admin import credentials
from firebase_admin import db
import traceback
import sys
import time
from firebase_admin import firestore
from create_interactive_html_report import make_interactive_html_report
session_type = 0
try:
    session_type = sys.argv[1]
except:
    print(traceback.format_exc())
cred = credentials.Certificate("servicekey.json")
# os.system("rm -rf AwsData")
try:
    firebase_admin.initialize_app(
        cred, {"databaseURL": "https://farmbase-b2f7e-31c0c.firebaseio.com/"}
    )
except:
    qysy = 1
dbF = firestore.client()
storage_client = storage.Client(project='farmbase-b2f7e')
bucket_name = "farmbase-b2f7e.appspot.com"
db_firestore = firestore.client()
disabledUIDs = [
                "snQYQZqQx3SmVbRztmEqYn5Mkcz2",
                "KZQ7TZIYXnXN0b07OtrL1hlyYij1",
                "CeMGYvLXrGR5ZThxZ46iV7vY8sa2",
                "TCXcp5VIsfhHZrh0nm2VsgBtcGy2",
                "mFFHQdEtiSbn2hbYQAwwoIdYVi02"
            ]
def upload_report(uid):
    bucket = storage_client.get_bucket(bucket_name)
    blob = bucket.blob(("PaidMonitoredFields/" + uid + "/interactive_report.html"))
    blob.upload_from_filename((uid + "/interactive_report.html"))
def instant_session():
    #todo
    a = 1
    while True:
        report_requests_obj = db_firestore.collection(u'InteractiveReportRequests').where('Processed','==',0).get()
        for temp_d in report_requests_obj:
            time_stamp = temp_d.id
            time_stamp = str(time_stamp)
            print(temp_d.id)
            print(temp_d.to_dict())
            temp_main_obj = temp_d.to_dict()
            uid = temp_main_obj.get("UID", None)
            make_interactive_html_report(uid)
            temp_main_obj["Processed"] = 1
            db_firestore.collection(u'InteractiveReportRequests').document(uid).set(temp_main_obj)
            upload_report(uid)
        time.sleep(60)


def daily_session():
    while True:
        uid_list = db.reference('PaidMonitoredFields').child('PMF').get(False, True)
        for uid, val in uid_list.items():
            if uid not in disabledUIDs:
                make_interactive_html_report(uid)
                upload_report(uid)
        time.sleep(86400)

if session_type == 0:
    instant_session()
elif session_type == 1:
    daily_session()
else:
    a = 1