LeetCode 1894. Find the student number that needs to be supplemented with chalk

python

class Solution:
    def chalkReplacer(self, chalk: List[int], k: int) -> int:
        
        sum_chalk = 0
        for i in range(len(chalk)):
            sum_chalk += chalk[i]
        
        if(k>sum_chalk):
            k = k%sum_chalk
        
        i = 0
        while(1):
            k -= chalk[i%len(chalk)]
            # print(k)
            if(k<0):
                break
            i +=1
        return  i%len(chalk)