import os

from models.user_preference import UserPreference, Preference
from models.base import with_session
from models.ratelimit import RateLimitWhitelist


LENGJING_ORG_DEV = "150002"
LENGJING_ORG_PROD = "1372813089209215500"


def sync_preference_dev():
    with with_session() as s:
        preference = UserPreference.load_by_org_id(LENGJING_ORG_DEV)
        preference.internal_api_enabled = True

        Preference.upsert_by_org_id(s, LENGJING_ORG_DEV, preference.to_json())

        for i in RateLimitWhitelist.get_all(s):
            preference = UserPreference.load_by_org_id(i.org_id)
            preference.rate_limit_per_day = i.count
            Preference.upsert_by_org_id(s, i.org_id, preference.to_json())

        preference = UserPreference.load_by_email_suffix("pingcap.com")
        preference.internal_api_enabled = True
        Preference.upsert_by_email_suffix(s, "pingcap.com", preference.to_json())


def sync_preference_prod():
    with with_session() as s:
        preference = UserPreference.load_by_org_id(LENGJING_ORG_PROD)
        preference.internal_api_enabled = True

        Preference.upsert_by_org_id(s, LENGJING_ORG_PROD, preference.to_json())

        for i in RateLimitWhitelist.get_all(s):
            preference = UserPreference.load_by_org_id(i.org_id)
            preference.rate_limit_per_day = i.count
            Preference.upsert_by_org_id(s, i.org_id, preference.to_json())

        preference = UserPreference.load_by_email_suffix("pingcap.com")
        preference.internal_api_enabled = True
        Preference.upsert_by_email_suffix(s, "pingcap.com", preference.to_json())


if __name__ == "__main__":
    if os.getenv("ENVIRONMENT") == "dev":
        sync_preference_dev()
    else:
        sync_preference_prod()