Grading students.py

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'gradingStudents' function below.
#
# The function is expected to return an INTEGER_ARRAY.
# The function accepts INTEGER_ARRAY grades as parameter.
#
final=[]
def gradingStudents(grades):
    for i in grades:
        if (i<38):
            final.append(i)
        else:
            if(i%5>=3):
                i=i+5-(i%5)
                final.append(i)
            elif(i%5<2):
                final.append(i)
            else:
                final.append(i)
    return final
                
            

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    grades_count = int(input().strip())

    grades = []

    for _ in range(grades_count):
        grades_item = int(input().strip())
        grades.append(grades_item)

    result = gradingStudents(grades)

    fptr.write('\n'.join(map(str, result)))
    fptr.write('\n')

    fptr.close()