用python解决 不是很理解“从主程序调用”

编写函数DisAllData，实现求出1000-9999之间的所有具有如下特点的四位数并显示，且把所有这样的四位数的和作为函数的返回值，在主程序中调用此函数。
它的平方根恰好就是它中间的两位数字（例如，2500开平方为50，恰为2500的中间两位）。

写个子函数，在主函数调用

import math
def DisAllData():
    sum = 0
    for n in range(1000,10000):
        k = math.sqrt(n)
        m = n//100%10*10 + n//10%10
        if k==m:
            sum = sum + n
            print(n)
     return sum
if __name__ == '__main__':
    sum  = DisAllData()
    print(sum)