[HackerRank-python] Repeated String

def repeatedString(s, n):
    # Write your code here
    
    s_len = len(s)
    
    if s_len == 1 and s == 'a':
        return n
    else:
        a_num = 0
        for i in s:
            if i == 'a':
                a_num +=1
                
        a_num = a_num * (n // s_len)
        if n % s_len != 0:
            for i in range(n % s_len):
                if s[i] == 'a':
                    a_num += 1
            
        return a_num