3
Nq iZ @ s d Z ddlZddlZddlZddlmZ dddddd d
d d d
ddddddddddddddgZeZ G d d deZ
G d d deZ d Z d Z
dd d!d d"d d"d d d"d d"d g
ZG d#d$ d$ZG d%d& d&Zed'Zed(Zed)Zed*Zed+\ZZZZZZZ d,d Z d-d Z d.d
Z d/d Z G d0d de!Z"G d1d de"Z#G d2d de"Z$G d3d4 d4Z%G d5d de#Z&G d6d de$Z'e# Z(e(j)Z*d7d Z+e(j,Z-e(j.Z.e(j/Z0e(j1Z2e(j3Z3e(j4Z5e(j6Z7e(j8Z9dDZ:d9Z;e:e;fd:d;ZZ>ej?e>d d j@ ZAd?d ZBd@dA ZCeDdBkreCejE dS )Ea$ Calendar printing functions
Note when comparing these calendars to the ones printed by cal(1): By
default, these calendars have Monday as the first day of the week, and
Sunday as the last (the European convention). Use setfirstweekday() to
set the first day of the week (0=Monday, 6=Sunday). N)repeatIllegalMonthErrorIllegalWeekdayErrorsetfirstweekday firstweekdayisleapleapdaysweekday
monthrange
monthcalendarprmonthmonthprcalcalendartimegm
month_name
month_abbrday_nameday_abbrCalendar TextCalendar HTMLCalendarLocaleTextCalendarLocaleHTMLCalendar
weekheaderc @ s e Zd Zdd Zdd ZdS )r c C s
|| _ d S )N)r
)selfr
r /usr/lib/python3.6/calendar.py__init__ s zIllegalMonthError.__init__c C s
d| j S )Nz!bad month number %r; must be 1-12)r
)r r r r __str__ s zIllegalMonthError.__str__N)__name__
__module__ __qualname__r r r r r r r s c @ s e Zd Zdd Zdd ZdS )r c C s
|| _ d S )N)r )r r r r r r s z IllegalWeekdayError.__init__c C s
d| j S )Nz7bad weekday number %r; must be 0 (Monday) to 6 (Sunday))r )r r r r r " s zIllegalWeekdayError.__str__N)r r! r" r r r r r r r s c @ sF e Zd Zdd edD Zejddd dd Zd d
Zd d Zd
S )_localized_monthc C s g | ]}t jd |d djqS )i r# )datetimedatestrftime).0ir r r
4 s z_localized_month. r c C s dS )N r )xr r r 5 s z_localized_month.c C s
|| _ d S )N)format)r r3 r r r r 7 s z_localized_month.__init__c s4 j | }t|tr& fdd|D S | jS d S )Nc s g | ]}| j qS r )r3 )r, f)r r r r. = s z0_localized_month.__getitem__..)_months
isinstanceslicer3 )r r- funcsr )r r __getitem__: s
z _localized_month.__getitem__c C s dS )N
r )r r r r __len__A s z_localized_month.__len__N) r r! r" ranger5 insertr r9 r; r r r r r( 2 s
r( c @ s6 e Zd Zdd edD Zdd Zdd Zdd Zd
S ) _localized_dayc C s g | ]}t jd d|d jqS )i r# )r) r* r+ )r, r- r r r r. H s z_localized_day. c C s
|| _ d S )N)r3 )r r3 r r r r J s z_localized_day.__init__c s4 j | }t|tr& fdd|D S | jS d S )Nc s g | ]}| j qS r )r3 )r, r4 )r r r r. P s z._localized_day.__getitem__..)_daysr6 r7 r3 )r r- r8 r )r r r9 M s
z_localized_day.__getitem__c C s dS )Nr? r )r r r r r; T s z_localized_day.__len__N)r r! r" r< r@ r r9 r; r r r r r> E s r> z%Az%az%Bz%br? c C s$ | d dko"| d dkp"| d dkS )z5Return True for leap years, False for non-leap years. r d i r )yearr r r r d s c C s@ | d8 } |d8 }|d | d |d | d |d | d S )zFReturn number of leap years in range [y1, y2).
Assume y1 <= y2.r# rA rB i r )Zy1Zy2r r r r i s c C s t j| ||j S )zTReturn weekday (0-6 ~ Mon-Sun) for year (1970-...), month (1-12),
day (1-31).)r) r* r )rC r
dayr r r r q s c C sL d| kodkn s t |t| |d}t| |tko@t| }||fS )zQReturn weekday (0-6 ~ Mon-Sun) and number of days (28-31) for
year, month.r# r/ )r r mdaysFebruaryr )rC r
day1ndaysr r r r
w s
c @ s e Zd ZdZd ddZdd Zdd ZeeeZd d
Z d d Z
d
d Z dd Z dd Z
dd Zdd Zd ddZd!ddZd"d d Zd S )#r zo
Base calendar class. This class doesn't do any formatting. It simply
provides data to subclasses.
r c C s
|| _ d S )N)r )r r r r r r s zCalendar.__init__c C s
| j d S )Nr? )
_firstweekday)r r r r getfirstweekday s zCalendar.getfirstweekdayc C s
|| _ d S )N)rI )r r r r r r s zCalendar.setfirstweekdayc c s* x$t | j| jd D ]}|d V qW dS )zt
Return an iterator for one week of weekday numbers starting with the
configured first one.
r? N)r< r )r r- r r r iterweekdays s zCalendar.iterweekdaysc c s t j||d}|j | j d }|t j|d8 }t jdd}xH|V y ||7 }W n tk
rf P Y nX |j|kr>|j | jkr>P q>W dS )z
Return an iterator for one month. The iterator will yield datetime.date
values and will always iterate through complete weeks, so it will yield
dates outside the specified month.
r# r? )daysN)r) r* r r Z timedelta
OverflowErrorr
)r rC r
r* rL Zonedayr r r itermonthdates s zCalendar.itermonthdatesc c s4 x.t | j||| jD ]\}}||d fV qW dS )z
Like itermonthdates(), but will yield (day number, weekday number)
tuples. For days outside the specified month the day number is 0.
r? N) enumerate
itermonthdaysr )r rC r
r- dr r r itermonthdays2 s zCalendar.itermonthdays2c c sf t ||\}}|| j d }td|E dH td|d E dH | j| | d }td|E dH dS )z
Like itermonthdates(), but will yield day numbers. For days outside
the specified month the day number is 0.
r? r Nr# )r
r r r< )r rC r
rG rH Z days_beforeZ
days_afterr r r rP s zCalendar.itermonthdaysc s. t | j|| fddtdt dD S )z
Return a matrix (list of lists) representing a month's calendar.
Each row represents a week; week entries are datetime.date values.
c s g | ]} ||d qS )r? r )r, r- )datesr r r. s z/Calendar.monthdatescalendar..r r? )listrN r< len)r rC r
r )rS r monthdatescalendar s zCalendar.monthdatescalendarc s. t | j|| fddtdt dD S )z
Return a matrix representing a month's calendar.
Each row represents a week; week entries are
(day number, weekday number) tuples. Day numbers outside this month
are zero.
c s g | ]} ||d qS )r? r )r, r- )rL r r r. s z/Calendar.monthdays2calendar..r r? )rT rR r< rU )r rC r
r )rL r monthdays2calendar s zCalendar.monthdays2calendarc s. t | j|| fddtdt dD S )z
Return a matrix representing a month's calendar.
Each row represents a week; days outside this month are zero.
c s g | ]} ||d qS )r? r )r, r- )rL r r r. s z.Calendar.monthdayscalendar..r r? )rT rP r< rU )r rC r
r )rL r monthdayscalendar s zCalendar.monthdayscalendar c s> fddt ttd D fddt dt D S )a'
Return the data for the specified year ready for formatting. The return
value is a list of month rows. Each month row contains up to width months.
Each month contains between 4 and 6 weeks and each week contains 1-7
days. Days are datetime.date objects.
c s g | ]} j |qS r )rV )r, r- )r rC r r r. s z.Calendar.yeardatescalendar..r/ c s g | ]} || qS r r )r, r- )monthswidthr r r. s r )r< JanuaryrU )r rC r[ r )rZ r r[ rC r yeardatescalendar s zCalendar.yeardatescalendarc s> fddt ttd D fddt dt D S )z
Return the data for the specified year ready for formatting (similar to
yeardatescalendar()). Entries in the week lists are
(day number, weekday number) tuples. Day numbers outside this month are
zero.
c s g | ]} j |qS r )rW )r, r- )r rC r r r. s z.Calendar.yeardays2calendar..r/ c s g | ]} || qS r r )r, r- )rZ r[ r r r. s r )r< r\ rU )r rC r[ r )rZ r r[ rC r yeardays2calendar s zCalendar.yeardays2calendarc s> fddt ttd D fddt dt D S )z
Return the data for the specified year ready for formatting (similar to
yeardatescalendar()). Entries in the week lists are day numbers.
Day numbers outside this month are zero.
c s g | ]} j |qS r )rX )r, r- )r rC r r r. s z-Calendar.yeardayscalendar..r/ c s g | ]} || qS r r )r, r- )rZ r[ r r r. s r )r< r\ rU )r rC r[ r )rZ r r[ rC r yeardayscalendar s zCalendar.yeardayscalendarN)r )rY )rY )rY )r r! r" __doc__r rJ r propertyr rK rN rR rP rV rW rX r] r^ r_ r r r r r s
c @ sj e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
dZ d ddZ
d ddZ d ddZ d!ddZ
d S )"r zr
Subclass of Calendar that outputs a calendar as a simple plain text
similar to the UNIX program cal.
c C s t | j||dd dS )z3
Print a single week (no newline).
)endN)print
formatweek)r theweekr[ r r r prweek
s zTextCalendar.prweekc C s |dkrd}nd| }|j |S )z*
Returns a formatted day.
r r0 z%2i)center)r rD r r[ sr r r formatday s zTextCalendar.formatdayc s dj fdd|D S )zA
Returns a single week in a string (no newline).
rb c 3 s | ]\}} j ||V qd S )N)rj )r, rQ wd)r r[ r r s z*TextCalendar.formatweek..)join)r rf r[ r )r r[ r re s zTextCalendar.formatweekc C s( |dkrt }nt}|| d| j|S )z4
Returns a formatted week day name.
N)r r rh )r rD r[ namesr r r
formatweekday s zTextCalendar.formatweekdayc s dj fdd j D S )z-
Return a header for a week.
rb c 3 s | ]} j |V qd S )N)rp )r, r- )r r[ r r rl . s z0TextCalendar.formatweekheader..)rm rK )r r[ r )r r[ r formatweekheader* s z TextCalendar.formatweekheaderTc C s" t | }|rd||f }|j|S )z0
Return a formatted month name.
z%s %r)r rh )r theyearthemonthr[ withyearri r r r formatmonthname0 s z TextCalendar.formatmonthnamer c C s t | j||||dd dS )z+
Print a month's calendar.
r0 )rc N)rd formatmonth)r rr rs wlr r r r 9 s zTextCalendar.prmonthc C s t d|}t d|}| j||d|d d }|j }|d| 7 }|| j|j 7 }|d| 7 }x4| j||D ]$}|| j||j 7 }|d| 7 }qnW |S )z@
Return a month's calendar string (multi-line).
r$ r# r?
)maxru rstriprq rW re )r rr rs rw rx ri weekr r r rv ? s
zTextCalendar.formatmonthr$ r# rY c s t d|}t d|}t d|}|d d d g }|j}|tj | ||d j |d| j|x@tj|D ]*\}} t|| d t ||d d d}
|d| fdd|
D } |t
| |j |d| fdd|
D } |t
| |j |d| t d d | D }
xvt|
D ]j}g }x>| D ]6}|t |krr|jd
n|jj || | qTW |t
| |j |d| qFW qW d
j
|S ) zC
Returns a year's calendar as a multi-line string.
r$ r# r? ry r: c 3 s | ]}j | d V qdS )FN)ru )r, k)colwidthr rr r r rl ` s z*TextCalendar.formatyear..c 3 s | ]
} V qd S )Nr )r, r~ )headerr r rl d s c s s | ]}t |V qd S )N)rU )r, calr r r rl h s r0 )rz appendreprrh r{ rq rO r^ r< min formatstringrU re rm )r rr rw rx cmvar- rowrZ ro ZheadersZheightjZweeksr r )r r r rr r
formatyearO s:
&
$
zTextCalendar.formatyearc C s t | j||||| dS )zPrint a year's calendar.N)rd r )r rr rw rx r r r r r pryeart s zTextCalendar.pryearN)T)r r )r r )r$ r# r} rY )r r r} rY )r r! r" r` rg rj re rp rq ru r rv r r r r r r r s
%c @ sj e Zd ZdZdddddddgZd d
Zd d Zd
d Zdd Zd ddZ d ddZ
d ddZ d dd Z dS )!r z4
This calendar returns complete HTML pages.
ZmonZtueZwedZthuZfriZsatZsunc C s" |dkr dS d| j | |f S dS )z/
Return a day as a table cell.
r z
z
%d
N)
cssclasses)r rD r r r r rj s zHTMLCalendar.formatdayc s dj fdd|D }d| S )z8
Return a complete week as a table row.
r0 c 3 s | ]\}} j ||V qd S )N)rj )r, rQ rk )r r r rl s z*HTMLCalendar.formatweek..z
%s
)rm )r rf ri r )r r re s zHTMLCalendar.formatweekc C s d| j | t| f S )z:
Return a weekday name as a table header.
z
%s
)r r )r rD r r r rp s zHTMLCalendar.formatweekdayc s$ dj fdd j D }d| S )z<
Return a header for a week as a table row.
r0 c 3 s | ]} j |V qd S )N)rp )r, r- )r r r rl s z0HTMLCalendar.formatweekheader..z
%s
)rm rK )r ri r )r r rq s z HTMLCalendar.formatweekheaderTc C s* |rdt | |f }n dt | }d| S )z5
Return a month name as a table row.
z%s %sz%sz.
%s
)r )r rr rs rt ri r r r ru s z HTMLCalendar.formatmonthnamec C s g }|j }|d |d || j|||d |d || j |d x*| j||D ]}|| j| |d qXW |d |d dj|S )z6
Return a formatted month as a table.
z@
ry )rt z
r0 )r ru rq rW re rm )r rr rs rt r r r| r r r rv s zHTMLCalendar.formatmonthrY c C s g }|j }t|d}|d |d |d||f xntttd |D ]Z}t|t|| d}|d x0|D ](}|d || j||d d
|d qlW |d qFW |d
dj|S )z?
Return a formatted year as a table of tables.
r# z?
ry z.
%s
r/ r: z
z
F)rt z
z
z
r0 )r rz r< r\ r rv rm )r rr r[ r r r- rZ r r r r r s
zHTMLCalendar.formatyear calendar.cssNc C s |dkrt j }g }|j}|d| |d |d |d |d| |dk r^|d| |d| |d |d
|| j|| |d |d d
j|j|dS )zB
Return a formatted year as a complete HTML page.
Nz$
zn
z
z
zC
z4
z Calendar for %d
z
z
z
z
r0 xmlcharrefreplace)sysgetdefaultencodingr r rm encode)r rr r[ cssencodingr r r r r formatyearpage s$ zHTMLCalendar.formatyearpage)T)T)rY )rY r N)
r r! r" r` r rj re rp rq ru rv r r r r r r r y s
c @ s$ e Zd Zdd Zdd Zdd ZdS )different_localec C s
|| _ d S )N)locale)r r r r r r s zdifferent_locale.__init__c C s" t jt j| _t jt j| j d S )N)_localeZ getlocaleLC_TIME oldlocale setlocaler )r r r r __enter__ s zdifferent_locale.__enter__c G s t jt j| j d S )N)r r r r )r argsr r r __exit__ s zdifferent_locale.__exit__N)r r! r" r r r r r r r r s r c @ s, e Zd ZdZd ddZdd Zd d d
ZdS )
r a
This class can be passed a locale name in the constructor and will return
month and weekday names in the specified locale. If this locale includes
an encoding all strings containing month and weekday names will be returned
as unicode.
r Nc C s& t j| | |d kr tj }|| _d S )N)r r r getdefaultlocaler )r r r r r r r s zLocaleTextCalendar.__init__c
C sB t | j. |dkrt}nt}|| }|d | j|S Q R X d S )Nrn )r r r r rh )r rD r[ ro namer r r rp s z LocaleTextCalendar.formatweekdayTc
C s8 t | j$ t| }|r$d||f }|j|S Q R X d S )Nz%s %r)r r r rh )r rr rs r[ rt ri r r r ru s
z"LocaleTextCalendar.formatmonthname)r N)T)r r! r" r` r rp ru r r r r r s
c @ s, e Zd ZdZd ddZdd Zd d d
ZdS )
r a
This class can be passed a locale name in the constructor and will return
month and weekday names in the specified locale. If this locale includes
an encoding all strings containing month and weekday names will be returned
as unicode.
r Nc C s& t j| | |d kr tj }|| _d S )N)r r r r r )r r r r r r r s zLocaleHTMLCalendar.__init__c
C s0 t | j t| }d| j| |f S Q R X d S )Nz
%s
)r r r r )r rD ri r r r rp ! s z LocaleHTMLCalendar.formatweekdayTc
C s6 t | j" t| }|r$d||f }d| S Q R X d S )Nz%s %sz.
%s
)r r r )r rr rs rt ri r r r ru & s
z"LocaleHTMLCalendar.formatmonthname)r N)T)r r! r" r` r rp ru r r r r r s
c C s* t | kotkn s t| | t_d S )N)MONDAYSUNDAYr r r )r r r r r 3 s rY r} c C s t t| || dS )z1Prints multi-column formatting for year calendarsN)rd r )colsr spacingr r r r3 G s r3 c s |d9 }|j fdd| D S )zEReturns a string formatted from n strings, centered within n columns.rb c 3 s | ]}|j V qd S )N)rh )r, r )r r r rl O s z formatstring..)rm )r r r r )r r r L s r i c C s^ | dd \}}}}}}t j||dj t | d }|d | }|d | } | d | }
|
S )zBUnrelated but handy function to calculate Unix timestamp from GMT.Nr} r# < )r) r* toordinal
_EPOCH_ORD) tuplerC r
rD ZhourZminutesecondrL ZhoursZminutesZsecondsr r r r V s c C s dd l }|j }|jd}|jd}|jddtddd |jd d
td d d |jd
dtddd |jddtddd |jddddd |jddd d d |jd d d d d |jd d!d"d3d$d% |jd&d'td(d) |jd*d'td+d) |j| d d }|jr|j r|jd, t j
d |j|jf}|j d#kr|jrFt |d-}nt
}|j}|d krdt j }t||jd.} t jjj}
|jd kr|
|jtjj jf| n6|jd kr|
|j|jf| n|jd/ t j
d n|jrt|d-}nt }t|j |j d0} |jd kr&|j | d1< |j | d2< |jd krJ|j tjj jf| } n2|jd krh|j |jf| } n|j!|j|jf| } t jj}
|jr| j"|j} t jjj}
|
| d S )4Nr ztext only argumentszhtml only argumentsz-wz--widthr$ z width of date column (default 2))typedefaulthelpz-lz--linesr# z)number of lines for each week (default 1)z-sz --spacingr} z"spacing between months (default 6)z-mz--monthsrY zmonths per row (default 3)z-cz--cssz calendar.csszCSS to use for page)r r z-Lz--localez.locale to be used from month and weekday namesz-ez
--encodingzencoding to use for outputz-tz--typetexthtmlzoutput type (text or html))r choicesr rC ?zyear number (1-9999))nargsr r r
z month number (1-12, text only)z/if --locale is specified --encoding is required)r )r r z incorrect number of arguments)rw rx r r )r r )#argparseArgumentParserZadd_argument_group add_argumentint
parse_argsr r errorr exitr r r r dictr stdoutbufferwriterC r r) r* Ztodayr
r r r[ linesr rZ r rv r ) r r parserZ textgroupZ htmlgroupZoptionsr r r Zoptdictr resultr r r main` s
r __main__ )Fr` r r) r r itertoolsr __all__
ValueErrorr r r r\ rF rE r( r> r r r r r< r ZTUESDAYZ WEDNESDAYZTHURSDAYZFRIDAYZSATURDAYr r r r r
objectr r r r r r r rJ r r rX r rg re r| rq r r rv r
r r r r Z _colwidthZ_spacingr3 r ZEPOCHr* r r r r r argvr r r r sn
up
h