ladok3 4.17__py3-none-any.whl → 4.19__py3-none-any.whl

ladok3/student.py

@@ -1,54 +1,75 @@
 import csv
 import ladok3.cli
 
+
 def print_student_data(student):
-  """Prints the student data, all attributes, to stdout."""
-  print(f"First name:   {student.first_name}")
-  print(f"Last name:    {student.last_name}")
-  print(f"Personnummer: {student.personnummer}")
-  print(f"LADOK ID:     {student.ladok_id}")
-  print(f"Alive:        {student.alive}")
-  print(f"Suspended:    {student.suspensions}")
+    """Prints the student data, all attributes, to stdout."""
+    print(f"First name:   {student.first_name}")
+    print(f"Last name:    {student.last_name}")
+    print(f"Personnummer: {student.personnummer}")
+    print(f"LADOK ID:     {student.ladok_id}")
+    print(f"Alive:        {student.alive}")
+    print(f"Suspended:    ", end="")
+    if any(map(lambda x: x.is_current, student.suspensions)):
+        print("YES")
+    else:
+        print("no")
+    if student.suspensions:
+        print(f"Suspenions:   ", end="")
+        for suspension in student.suspensions:
+            print(f"{suspension}", end="\n              ")
+        print()
+
+
 def print_course_data(student, args):
-  """Prints the courses"""
-  print("Courses:")
-  for course in student.courses(code=args.course):
-    print(f"{course}")
-    if args.results:
-      for result in course.results():
-        print(f"  {result}")
+    """Prints the courses"""
+    print("Courses:")
+    for course in student.courses(code=args.course):
+        print(f"{course}")
+        if args.results:
+            for result in course.results():
+                print(f"  {result}")
+
 
 def add_command_options(parser):
-  student_parser = parser.add_parser("student",
-    help="Shows a student's information in LADOK",
-    description="""
+    student_parser = parser.add_parser(
+        "student",
+        help="Shows a student's information in LADOK",
+        description="""
     Show a student's information in LADOK.
     Shows information like name, personnummer, contact information.
-    """
-  )
-  student_parser.set_defaults(func=command)
-  student_parser.add_argument("id",
-    help="The student's ID, either personnummer or LADOK ID"
-  )
-  student_parser.add_argument("-c", "--course",
-    nargs="?", const=".*",
-    help="A regular expression for which course codes to list, " \
-      "use no value for listing all courses."
-  )
-  student_parser.add_argument("-r", "--results",
-    action="store_true", default=False,
-    help="Set to include results for each course listed."
-  )
+    """,
+    )
+    student_parser.set_defaults(func=command)
+    student_parser.add_argument(
+        "id", help="The student's ID, either personnummer or LADOK ID"
+    )
+    student_parser.add_argument(
+        "-c",
+        "--course",
+        nargs="?",
+        const=".*",
+        help="A regular expression for which course codes to list, "
+        "use no value for listing all courses.",
+    )
+    student_parser.add_argument(
+        "-r",
+        "--results",
+        action="store_true",
+        default=False,
+        help="Set to include results for each course listed.",
+    )
+
 
 def command(ladok, args):
-  try:
-    student = ladok.get_student(args.id)
-    student.alive
-  except Exception as err:
-    ladok3.cli.err(-1, f"can't fetch student data for {args.id}: {err}")
+    try:
+        student = ladok.get_student(args.id)
+        student.alive
+    except Exception as err:
+        ladok3.cli.err(-1, f"can't fetch student data for {args.id}: {err}")
 
-  print_student_data(student)
+    print_student_data(student)
 
-  if args.course:
-    print()
-    print_course_data(student, args)
+    if args.course:
+        print()
+        print_course_data(student, args)