Minor changes

1 files changed, 19 insertions, 0 deletions

diff --git a/main.py b/main.py
index a4912ec..bfc63b8 100755
--- a/main.py
+++ b/main.py
@@ -19,6 +19,13 @@ def formatday(d):
     return d.strftime(i["output"]["date_format"])
 def getweek(d):
     return d.isocalendar().week + i["output"]["week_drift"]
+def getgloballectureslot(d):
+    dname=d.strftime("%A").lower()
+    if dname in i["semester"]["lectures"]["slots"].keys():
+        start=datetime.strptime(i["semester"]["lectures"]["slots"][dname]["start"], "%I%p")
+        end=datetime.strptime(i["semester"]["lectures"]["slots"][dname]["end"], "%I%p")
+        (start,end)
+    return None
 def getassign(d):
     val=""
     for a in i["assignments"]:
@@ -30,12 +37,24 @@ def getassign(d):
             val+=str(a)
     return val
 def getlecture(d):
+    slot=getgloballectureslot(d)
     for l in i["lectures"]:
         date=parse_date(i["lectures"][l]["date"])
         if d==date:
+            # Time
+            startT, endT = (None, None)
+            if "slot" in i["lectures"][l]:
+                startT=datetime.strptime(i["lectures"][l]["slot"]["start"], "%I%p")
+                endT=datetime.strptime(i["lectures"][l]["slot"]["end"], "%I%p")
+            elif slot is not None:
+                startT, endT=slot
+            # Content
             text=textwrap.fill(i["lectures"][l]["name"],o["text_wrap"])
             if o["show_lecturers"]:
                 text+="\n"+textwrap.fill("("+i["lectures"][l]["who"]+")",o["text_wrap"])
+            if (startT,endT) != (None, None):
+                timeT=textwrap.fill(startT.strftime(i["output"]["time_format"])+"-"+endT.strftime(i["output"]["time_format"]),o["text_wrap"])
+                text=timeT+"\n"+text
             return text
     return ""
 def getevents(d):