Question

我使用dateutil根据输入文件生成计划。

我的配置文件如下所示：

[alarm1]
time = 8:00
days = MO

[alarm2]
time = 9:00
days = TU, WE

[alarm3]
time = 22:00
days = MO, WE, FR
etc..

我使用dateutil根据此生成计划。

这里是一些示例代码，为了简洁起见，我省略了控制流程和导入代码。

from dateutil.rrule import *
from dateutil.parser import *
import datetime
import configparser


alarms = rruleset()
time = parse(alarm_parser.get(alarm_section, 'time')).time()
date = datetime.date.today()
alarm_time= datetime.datetime.combine(date, time)
days = '(' + alarm_parser.get(alarm_section, 'days') + ',)'
alarm_days = eval(days)

alarms.rrule(rrule(WEEKLY, dtstart = alarm_datetime, byweekday = alarm_days,  count = 1))

此代码现在可以使用eval但是，eval显然存在风险。

我已经尝试了rrulestr（），但是这个命令似乎并不接受byweekday =作为参数的一部分。

如何在不使用eval的情况下将配置文件中的字符串转换为rrule可以使用的元组？我尝试过ast.literal_eval，但这给了我以下错误

ValueError: malformed node or string: <_ast.Name object at 0xb68e2430>

Answer 1

我需要Paul的简单代码的稍微复杂一点的版本（可以处理带括号的偏移量，例如“ FR（+2）”）。这是我想出的：

import re
from dateutil.rrule import MO, TU, WE, TH, FR, SA, SU

_daymap = {"MO": MO, "TU": TU, "WE": WE, "TH": TH, "FR": FR, "SA": SA, "SU": SU}
_dayre = re.compile(r"(MO|TU|WE|TH|FR|SA|SU)(?=\(([-+]?[0-9]+)\))?")


def parsedays(daystr):
    for day in daystr.split(","):
        m = _dayre.match(day.strip())
        if m:
            weekday, offset = m.groups()
            result = _daymap[weekday]
            if offset:
                result = result(int(offset))
            yield result


def dumpdays(days):
    return ",".join(str(d) for d in days)

使用Python 3.7.3测试

Answer 2

这样做的简单方法是：

from dateutil.rrule import MO, TU, WE, TH, FR, SA, SU

daymap = {'MO': MO, 'TU': TU, 'WE': WE, 'TH': TH, 'FR': FR, 'SA': SA, 'SU': SU}
daystr = 'TU, TH'

days = [daymap[day.strip()] for day in daystr.split(',')]

但是，您也可以使用rrulestr（非BYDAY）参数BYWEEKDAY来完成此操作：

from dateutil.rrule import rrulestr
base_rrstr = ('DTSTART:{alarm_datetime:%Y%m%dT%H%M%S}\n' +
              'FREQ={freq_str};COUNT={count_str};BYDAY={day_str}')

alarm_datetime = datetime.datetime.today()
freq_str = "WEEKLY"
count_str = "1"
day_str = "TU,TH"

crr = rrulestr(base_rrstr.format(alarm_datetime=alarm_datetime,
                                 freq_str=freq_str,
                                 count_str=count_str,
                                 day_str=day_str))

如何在不使用eval

2 个答案: