B `@sddlmZddlmZddlmZddlZddlZddlZddddgZdd dZdd dZ d dZ ddZ ddZ ddZ dddZddZddZdddZdS))print_function)division)absolute_importNadd_cron erase_cron remove_cron add_cron_taskTc Cs|dkrtttdd}|dkr8tttdd}yt|||||||} Wn tk rptjdYnXt || |dS)a Add new cron task into crontab schedule iff this task or command wasn't already existed in the cron-file. :param str file_name: Name of cron-file in /etc/cron.d :param minute: Integer or char 'r' if to set random minute :param hour: Integer or char 'r' if to set random hour :param int, str day: Day number :param int, str month: Month number :param int, str day_of_week: Day of week number :param str user: Under what user do run command :param str command: What command do run :param bool check_command: If it is False, check that whole cron-task line already exists in crontab, check that command string exists instead. Default is True, check a command string. rr;z"Can not add task with wrong syntaxN) introundrandomZuniformformat_cron_task TypeErrorsysstderrwriter) file_nameminutehourdaymonth day_of_weekusercommand check_commandZ cron_taskr crontab.pyrsFc Csfd}y,td|d}t|||s.|d|Wn(ttfk rX|dk rT|dSX|dS)a Add new cron task in cron format iff this task or command in this task wasn't already existed in the cron-file. :param str file_name: Name of cron-file in /etc/cron.d :param str task: Cron task in format "min hour day mon d_of_w user command" :param bool check_command: If it is False, check that whole cron-task line already exists in crontab, check that command string exists instead. Default is False, check a whole cron-task string. Nz /etc/cron.d/za+z%s FT)open is_in_cronrIOErrorOSErrorclose)rtaskrfrrrr-s  c Cs0ytd|Wnttfk r*YnXdS)z^ Remove cron-file from fs :param str file_name: Name of cron-file in /etc/cron.d z /etc/cron.d/N)osremover"r!)rrrrrCsc Csjd}ytd|d}Wn>ttfk rT}ztjd|t|fWdd}~XYnX|dk rf|dS)zZ Make cron-file empty :param str file_name: Name of cron-file in /etc/cron.d Nz /etc/cron.d/wz)Can not erase crontab file %s because %s )rr!r"rrrstrr#)rr%errrrrrOs"c Cs8|||||||f}x|D]}|dkrtdqWd|S)a Build cron-task string in the cron format :param minute: Integer or char 'r' if to set random minute :param hour: Integer or char 'r' if to set random hour :param int day: Day number :param int month: Month number :param int day_of_week: Day of week number :param str user: Under what user do run command :param str command: What command do run :return: Cron-task in the cron format :rtype: str NzWrong schedule for cron taskz%2s %2s %2s %2s %2s %10s %s)r) rrrrrrrZ argumentsargrrrr_s   rcCs |ddS)z Split cron task string into cron task parts :param str task: Cron-task string in the cron format :return: List of cron-task parts :rtype: list of str N)split)r$rrrparse_cron_tasktsr.c csfx`dd|DD]N}y|r$t|}n tt|}Wn$tk rVtjd|YqX|VqWdS)a< Returns iterator through crontab tasks :param iterable crontab: Iterator with crontab tasks' strings :param bool get_parsed: If it is True, return crontab task as list of task's parts return crontab task as a string instead :return: Crontab task :rtype: str :rtype: list of str css|]}|VqdS)N)strip).0srrr sz#get_task_in_cron..zWrong crontab task syntax: %s N)r.rrrrr)Zcrontab get_parsedZcron_ttrrrget_task_in_crons  r5cCs&x t|D]}||krdSqWdS)a Find first occurence of task in cron-file if it has :param str task: Cron-task in cront format to compare with :param file fd: File descriptor of opened cron file :return: True if such a task is already existed in cron-file, False instead :rtype: bool TF)r5 readlines)r$fdr4rrris_task_in_crons r8cCs:t|d}x(t|ddD]}|d|krdSqWdS)a :param str task: Task with command to looking for :param file fd: File descriptor of opened cron file :return: True if such a command is already existed in cron-file, False instead :rtype: bool Find first occurence of command in cron-file if it has T)r3F)r.r5r6)r$r7rr4rrris_command_in_crons  r:cCs|rt||St||S)a Find first occurence of command or task in cron-file if it has :param str task: Task or command to looking for :param file, BinaryIO fd: File descriptor of opened cron file :param bool check_command: If it is True, check command occurence, check task occurence instead :return: True if such a command or task is already existed in cron-file, False instead :rtype: bool )r:r8)r$r7rrrrr s  r )T)F)F)F)Z __future__rrrr&rrZ__ALL__rrrrrr.r5r8r:r rrrrs