số lượng Ước số của n python

from math import sqrt
def uc(n):
  d = 0
  for i in range(1, int(sqrt(n))+1):
    if n % i == 0:
      d+= 1
      if i ** 2 != n:
          d+=1
  return d

print(uc(int(input())))

import math

n = int(input('Nhập số n: '))
count = 0

for i in range(1, int(math.sqrt(n))+1):
    if n % i == 0:
        count += 2 if i != n // i else 1

print('Số lượng ước số của', n, 'là:', count)