pythonでカレンダー作りました。

from datetime import date,timedelta
dd=["日","月","火","水","木","金","土"]
print("yyyy")
y=int(input())
print("mm")
m=int(input())

#y=2024
#m=6

d1=date(y,m,1)
day_index=date.weekday(d1)+1
if day_index>6:
    day_index=day_index-7
#print(day_index)
if m==12:
    d2=date(y+1,1,1)
else:
    d2=date(y,m+1,1)
d2=d2-timedelta(days=1)
#print(d1,d2)
dd1=d1.day
dd2=d2.day
#print(dd1,dd2)
print(" \033[31m日\033[37m 月 火 水 木 金 \033[34m土\033[37m")
for i in range(day_index):
    print("   ",end="")
count=day_index
n=1
while(True):
    if count==0:
        print('\033[31m',end="") #RED
    else:
        if count==6:
            print('\033[34m',end="")#BLUE
        else:
            print('\033[37m',end="")#WHITE
    if n<10:
        nn=" 0"+str(n)
    else:
        nn=" "+str(n)
    print(nn,end="")
    n=n+1
    if n>dd2:
        break
    count=count+1
    if count==7:
        count=0
        print()
print('\033[37m',end="")#WHITE

西暦年と月を入力すると1か月分のカレンダーが表示されます。
素人なもので、こてこてのプログラムです。

出力結果（実際の土は青色、日は赤色です。）

yyyy

2024
mm

12
 日 月 火 水 木 金 土
 01 02 03 04 05 06 07
 08 09 10 11 12 13 14
 15 16 17 18 19 20 21
 22 23 24 25 26 27 28
 29 30 31