Commit 2cf07028 authored by 吴升宇's avatar 吴升宇

fix pgsql

parent c9ca7e35
# coding: utf-8
from datetime import timedelta, datetime
import calendar
from airflow import DAG
from airflow.operators.postgres_operator import PostgresOperator
......@@ -23,21 +24,19 @@ dag = DAG(
schedule_interval='0 0 1 * * 0', # 每月一号零点执行一次
)
import datetime, calendar
# 获取当前日期
now_time = datetime.datetime.now()
now_time = datetime.now()
# 获取当前时间的星期数和月数
week, days_num = calendar.monthrange(now_time.year, now_time.month)
# 获取本月的最后一天
end_day_in_mouth = now_time.replace(day=days_num)
# 获取下月的第一天
start_day = end_day_in_mouth + datetime.timedelta(days=1)
start_day = end_day_in_mouth + timedelta(days=1)
month = start_day.strftime('y%Ym%m')
# 获取下月的星期数和月数
week, days_num = calendar.monthrange(start_day.year, start_day.month)
# 获取下个月的最后一天
end_day = start_day + datetime.timedelta(days=(days_num-1))
end_day = start_day + timedelta(days=(days_num-1))
end_str = str(end_day.date())
start_str = str(start_day.date())
......@@ -48,6 +47,7 @@ CREATE TABLE events_{month} partition of events
""".format(month=month, start_str=start_str, end_str=end_str)
print(sql1)
sql2 = """
ALTER TABLE events_{month} ADD CONSTRAINT pk_events_{month} PRIMARY KEY(id);
""".format(month=month)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment