bigdataAPI/crond_script/crond_weather.py

import requests
import sqlite3
import json
import os
import sys
from weather import all_city
import logging
import django
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'bigdataAPI.settings')
django.setup()
from apps.Weather.models import District, DayData, ServerDayData
# 配置日志级别和格式
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 创建一个文件处理器，指定日志文件名和写入模式
os.mkdir("/data/weather/") if not os.path.exists("/data/weather/") else None
file_handler = logging.FileHandler('/data/weather/app.log')
file_handler.setLevel(logging.INFO)

# 创建一个格式化程序，用于定义日志消息的显示方式
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
file_handler.setFormatter(formatter)

# 将文件处理器添加到根日志记录器中
logging.getLogger('').addHandler(file_handler)

app_id = "69334222"
app_secret = "2u4bHXHD"

province_dict = {
"11": '北京市',
"12": '天津市',
"13": '河北省',
"14": '山西省',
"15": '内蒙古自治区',
"21": '辽宁省',
"22": '吉林省',
"23": '黑龙江省',
"31": '上海市',
"32": '江苏省',
"33": '浙江省',
"34": '安徽省',
"35": '福建省',
"36": '江西省',
"37": '山东省',
"41": '河南省',
"42": '湖北省',
"43": '湖南省',
"44": '广东省',
"45": '广西壮族自治区',
"46": '海南省',
"50": '重庆市',
"51": '四川省',
"52": '贵州省',
"53": '云南省',
"54": '西藏自治区',
"61": '陕西省',
"62": '甘肃省',
"63": '青海省',
"64": '宁夏回族自治区',
"65": '新疆维吾尔自治区'
}

db = "/data/weather/weather.db"
def init_data():
    # 导入全部的城市
    for k in all_city:
        District.objects.get_or_create(
            id=k[0],
            defaults={
                "id": k[0],
                "city": k[1],
                "pid": k[2]
            }
        )
    

def main():
    district = District.objects.all() 
    if not district:
        init_data()
    logging.warning("up...")
    conn1 = sqlite3.connect(db)
    cursor1 = conn1.cursor()
    sql = """
        select * from district where id > 65
    """
    cursor1.execute(sql)
    results = cursor1.fetchall()

    results = District.objects.filter(id__gt = 65)
    # 更新数据
    for row in results:
        temp = row.city   # 东城区
        id = row.id       # 110101000000
        pid = row.pid     # 110100000000
        if len(str(pid)) == 2:
            # 处理 省 市 样例
            city = temp 
            district = ""
        else:
            te = District.objects.filter(pid=pid).first()
            city = te[0] 
            district = temp
            if city == "市辖区":
                city = district
                district = ""
        province_id = (str(row.id))[:2]
        province = province_dict.get(province_id)
        try:
            today_url = f"https://v0.yiketianqi.com/api?unescape=1&version=v62&appid={app_id}&appsecret={app_secret}&adcode={id}"
            today_response = requests.get(today_url, timeout=5)
            today_data = json.loads(today_response.text)
            if "errcode" not in today_data.keys():
                cityid = today_data.get("cityid", "")
                DayData.objects.update_or_create(
                    cityid=cityid,
                    defaults={
                        "cityid": cityid,
                        "province": province,
                        "city": city,
                        "district": district,
                        "content": str(today_data)
                    }
                )
            server_day_url = f"https://v0.yiketianqi.com/api?unescape=1&version=v91&appid={app_id}&appsecret={app_secret}&adcode={id}&ext=hours,aqi,life"
            server_day_response = requests.get(server_day_url, timeout=5)
            server_day_data = json.loads(server_day_response.text)
            if "errcode" not in server_day_data.keys():
                cityid = today_data.get("cityid", "")
                ServerDayData.objects.update_or_create(
                    cityid=cityid,
                    defaults={
                        "cityid": cityid,
                        "province": province,
                        "city": city,
                        "district": district,
                        "content": str(today_data)
                    }
                )
        except Exception as e:
            print(e)
    logging.warning("over...")


if __name__ == "__main__":
    main()