德国大神开发的LaTeX日历生成工具

1. # -*- coding: utf-8 -*-
   
2. 
   
3. import calendar
   
4. import datetime
   
5. 
   
6. def number_of_weeks(year, month):
   
7.     """
   
8.         Returns a tupel with the ISO no of the first and week and the no of weeks
   
9.     """
   
10.     tup_month_days = calendar.monthrange(year, month)
    
11.     first = datetime.date(year, month, 1)
    
12.     last = datetime.date(year, month, tup_month_days[1])
    
13.     first_week = first.isocalendar()[1]
    
14.     last_week = last.isocalendar()[1]
    
15.     return (first_week, last_week, last_week-first_week+1)
    
16. 
    
17. def gen_cal_latex(year, month):
    
18.     """
    
19.         https://stackoverflow.com/questions/9459337/assign-value-to-an-individual-cell-in-a-two-dimensional-python-array
    
20.     """
    
21.     c = calendar.TextCalendar()
    
22.     week_month_first, week_month_last, no_of_weeks = number_of_weeks(year, month)
    
23. 
    
24.     # generate calendar list, using tupel as a key   
    
25.     m = {(i, j):' ' for i in range(no_of_weeks) for j in range(7)}
    
26. 
    
27.     for tupel_date in c.itermonthdays4(year, month):
    
28.         t_year, t_month, t_day, t_weekday = tupel_date
    
29.         # use only dates inside the required month
    
30.         if t_month == month:
    
31.             temp_date = datetime.date(t_year, t_month, t_day)
    
32.             # check in which week we are with the current date
    
33.             # to get index for the list
    
34.             week_no = temp_date.isocalendar()[1]
    
35.             m[week_no % week_month_first, t_weekday] = t_day
    
36. 
    
37.     print(r'\begin{tabular}{rrrrrrr}')
    
38.     print(r'Mo & Di & Mi & Do & Fr & Sa & So \\')
    
39.     for i in m:
    
40.         if i[1] < 6:
    
41.             print('{0} &'.format(m[i]), end='')
    
42.         else:
    
43.             print('{0}'.format(m[i]),end='')
    
44.         if i[1] == 6:
    
45.             print(r'\\')
    
46.     print(r'\end{tabular}')
    
47. 
    
48. gen_cal_latex(2020, 4)

复制代码