Utils.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. # Author renyupeng
  2. # coding=utf-8
  3. # @Time : 2021/9/15 11:06 上午
  4. # @Site :
  5. # @File : Utils.py
  6. # @Software: PyCharm
  7. # @contact: renyupeng@c-top.com.cn
  8. import argparse
  9. import datetime
  10. from datetime import timedelta, date
  11. import calendar
  12. class Utils:
  13. """
  14. 参数获取工具类
  15. """
  16. @staticmethod
  17. def get_args():
  18. parser = argparse.ArgumentParser(description='')
  19. parser.add_argument('-start_time', type=str, default=None, help="TX_DT")
  20. parser.add_argument('-end_time', type=str, default=None, help="TX_DT")
  21. args = parser.parse_args()
  22. start_time = args.start_time
  23. end_time = args.end_time
  24. return start_time, end_time
  25. @staticmethod
  26. def get_task_args():
  27. parser = argparse.ArgumentParser(description='')
  28. parser.add_argument('-date_time', type=str, default=None, help="TX_DT")
  29. parser.add_argument('-day_count', type=int, default=None, help="DAY")
  30. args = parser.parse_args()
  31. date_time = args.date_time
  32. day_count = args.day_count
  33. return date_time, day_count
  34. @staticmethod
  35. def get_month_range(start_date=None):
  36. if start_date is None:
  37. start_date = date.today().replace(day=1)
  38. _, days_in_month = calendar.monthrange(start_date.year, start_date.month)
  39. end_date = start_date + timedelta(days=days_in_month)
  40. return start_date, end_date
  41. @staticmethod
  42. def get_days():
  43. yesterday = (datetime.date.today() + datetime.timedelta(-1)).strftime("%Y-%m-%d")
  44. ll = []
  45. day = timedelta(days=1)
  46. startday, endday = Utils.get_month_range()
  47. print(str(startday))
  48. while startday < endday:
  49. ll.append(str(startday))
  50. startday += day
  51. print('----',ll)
  52. lt = ll.index(yesterday)
  53. return lt
  54. if __name__ == '__main__':
  55. print(Utils.get_days())