From eed71007157010b4657d15656f5936a3a715ccb0 Mon Sep 17 00:00:00 2001
From: DSK <sureshdsk91@gmail.com>
Date: Sun, 3 Oct 2021 12:43:08 +0530
Subject: [PATCH] added example for calendar layout using columns and tables

---
 examples/columns2.py | 56 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 examples/columns2.py

diff --git a/examples/columns2.py b/examples/columns2.py
new file mode 100644
index 00000000..4e97eeba
--- /dev/null
+++ b/examples/columns2.py
@@ -0,0 +1,56 @@
+"""
+Builds calendar layout using Columns and Tables.
+Usage:
+python columns2.py [YEAR]
+Example:
+python columns2.py 2021
+"""
+import argparse
+
+import calendar
+from datetime import datetime
+
+from rich.table import Table
+from rich.columns import Columns
+from rich import print
+
+
+def display_calendar(year=None):
+    
+    today = datetime.today()
+    year = int(year or today.year)
+    cal = calendar.Calendar()
+
+    tables = []
+    for month in range(1,13):
+
+        table = Table(title=f"""{calendar.month_name[month]} {year}""", style="green")
+
+        for wd in cal.iterweekdays():
+            table.add_column("{:.3}".format(calendar.day_name[wd]))
+
+        month_days = cal.monthdayscalendar(year, month)
+        for weekdays in month_days:
+            days = []
+            for day in weekdays:
+                today_tuple = today.day, today.month, today.year
+                
+                if day and (day, month, year) == today_tuple :
+                    _day = "[bold underline red]%s[/]" % day   
+                else:
+                    _day = "%s" % ('' if day == 0 else day) 
+                days.append(_day)
+            table.add_row(*days)
+
+        tables.append(table)
+
+    columns = Columns(tables, equal=True, expand=True)
+    print(columns)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Rich calendar')
+    parser.add_argument('year', metavar='year', type=int, default=None)
+    args = parser.parse_args()
+
+    display_calendar(args.year)