数据仓库日期维表怎么做

最近终于有点时间了,东拼西凑的设计了一个日期维表,也是数仓设计时比较常用的。表结构如下:

dt string COMMENT '日历日期',
lunar_dt string COMMENT '农历日期',
calendar_year int COMMENT '日历年',
quarter_of_year int COMMENT '季度',
month_of_year int COMMENT '月份',
week_of_year int COMMENT '一年第几周',
dt_of_week int COMMENT '一周第几天',
dt_of_month int COMMENT '一个月第几天',
dt_of_year int COMMENT '一年第几天',
is_last_day_of_month boolean COMMENT '是否一个月最后一天',
is_weekend boolean COMMENT '是否周末日',
is_holiday boolean COMMENT '是否节假日',
holiday_cnt tinyint COMMENT '节日天数',
holiday_type tinyint COMMENT '日历天类型:1:工作日;2:周末(和假期连着按假期,因节假日调休上班按工作日算:3:元旦 4:春节 5:清明节 6:劳动节:7:端午节 8:中秋节 9:国庆'

实际落表代码没有写,只写了数据获取部分,具体实现代码如下,欢迎大佬指点~

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
import requests
import json
import datetime
import time
import calendar
import lunardate
import csv

reload(sys)                      # reload 才能调用 setdefaultencoding 方法
sys.setdefaultencoding('utf-8')

holiday_dict = {}

# 判断二月闰月
def isLeapYear(year):
    return calendar.monthrange(year, 2)[1] == 29

# 季度信息
def quarter(month):
    return (month - 1) / 3 + 1

# 周信息
def get_week_of_year(dt):
    return dt.isocalendar()[1]

def get_day_of_year(dt):
    return dt.timetuple().tm_yday

def get_day_of_week(day):
    return day.weekday() + 1 % 7
# 返回是否最后一天
def last_day_of_month(year, month, day):
    return day == calendar.monthrange(year, month)[1]

def get_is_weekend(day):
    return day.weekday() == 5 or day.weekday() == 6

def get_holiday_dict(year, month):
    params = {
        'resource_id': 6018,
        'ie': 'utf8',
        'oe': 'utf8',
        'format': 'json',
        'tn': 'baidu'
    }
    query_arg = "%d年%d月"
    params['query'] = query_arg % (year, month)
    res = requests.get(url='https://sp0.baidu.com/8aQDcjqpAAV3otqbppnN2DJv/api.php?', params=params).text
    # 取括号内的
    res = res[res.find("{"): res.rfind("}") + 1]

    js = json.loads(res,encoding='utf-8')

    holiday_list = js.get('data')[0].get('holiday')

    for i in holiday_list:
        list_list = i.get('list')
        name = i.get('name')
        day_cnt = len(list_list)
        for x in list_list:
            status = x.get('status')
            if status == 2:
                day_cnt -= 1

        for j in list_list:
            dt = j.get('date')
            status = j.get('status')
            item_info = (status,name,day_cnt)
            holiday_dict[dt] = item_info

# 获取阴历日期
def get_lunar_dt(dt):
    ymd = dt.split('-')
    lunar = lunardate.LunarDate.fromSolarDate(int(ymd[0]), int(ymd[1]), int(ymd[2]))
    return "%d-%02d-%02d" % (lunar.year, lunar.month, lunar.day)


def get_is_holiday(day):
    dt = day.strftime("%Y-%-m-%-d")
    item = holiday_dict.get(dt)
    if item is not None :
        return int(item[0]) == 1
    else:
        return False

def is_work_day(day,is_weekend):
    dt = day.strftime("%Y-%-m-%-d")
    item = holiday_dict.get(dt)
    if item is not None :
        # 如果在holiday列表里但是状态等于2
        return int(item[0]) == 2
    else:
        return not (is_weekend)

def get_holiday_cnt(day):
    dt = day.strftime("%Y-%-m-%-d")
    item = holiday_dict.get(dt)
    if item is not None and int(item[0]) == 1:
        return int(item[2])
    else:
        return 0

def get_holiday_type(day,is_weekend):
    dt = day.strftime("%Y-%-m-%-d")
    item = holiday_dict.get(dt)
    if item is not None :
        if int(item[0]) == 1:
            if str(item[1])  == '元旦'  :
                return 3
            elif str(item[1])  == '春节' :
                return 4
            elif str(item[1]) == '清明节' :
                return 5
            elif str(item[1])  == '劳动节' :
                return 6
            elif str(item[1])  == '端午节' :
                return 7
            elif str(item[1])  == '中秋节' :
                return 8
            elif str(item[1])  == '国庆节':
                return 9
            else:
                return 99
        else:
            return 1

    else:
        if is_weekend:
            return 2
        else:
            return 1
    return 1

def write_csv(data_row):
    path  = "aa.csv"
    with open(path,'a+') as f:
        csv_write = csv.writer(f)
        csv_write.writerow(data_row)

if __name__ == '__main__':
    start_date = datetime.datetime(2020,1,1)
    end_date = datetime.datetime(2020,1,31)
    day = start_date
    get_holiday_dict(2020,1)

    while(day <= end_date):
        dt = day.strftime("%Y-%m-%d")
        lunar_dt = get_lunar_dt(dt)

        calendar_year = day.year
        quarter_of_year = quarter(day.month)
        month_of_year = day.month
        week_of_year = get_week_of_year(day)
        dt_of_week = get_day_of_week(day)
        dt_of_month = day.day
        dt_of_year = get_day_of_year(day)
        is_last_day_of_month = last_day_of_month(calendar_year,month_of_year,dt_of_month)
        is_weekend = get_is_weekend(day)
        is_holiday = get_is_holiday(day)
        holiday_type = get_holiday_type(day,is_weekend)
        holiday_cnt = get_holiday_cnt(day)
        data_raw = [dt,lunar_dt,calendar_year,quarter_of_year,month_of_year,week_of_year,dt_of_week,dt_of_month,dt_of_year,is_last_day_of_month,is_weekend,is_holiday,holiday_type,holiday_cnt]
        print data_raw
        write_csv(data_raw)
        # print '日期:'+dt+ '年份'+str(calendar_year)+'Q'+str(quarter_of_year)+'月份'+str(month_of_year)+'周'+str(week_of_year)+'星期'+str(dt_of_week)+' 月的第'+str(dt_of_month)+' 年的第'+str(dt_of_year)+str(is_weekend)

        day = day + datetime.timedelta(days=1)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,454评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,553评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,921评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,648评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,770评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,950评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,090评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,817评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,275评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,592评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,724评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,409评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,052评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,815评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,043评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,503评论 2 361
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,627评论 2 350

推荐阅读更多精彩内容