atoi() 函数

一个账号

这个代码还有很多地方需要完善，欢迎大家一起交流

1. import math
   
2. 
   
3. def atoi(x=0, base=10) -> int:
   
4.     if base < 2 and base != 0 or base > 36:
   
5.         raise ValueError("atoi() base must be >= 2 and <= 36, or 0")
   
6.    
   
7.     if isinstance(x, int):
   
8.         if base != 10:
   
9.             raise TypeError("atoi() can't convert non-string with explicit base")
   
10.             
    
11.         return x
    
12. 
    
13.     if isinstance(x, float):
    
14.         if base != 10:
    
15.             raise TypeError("atoi() can't convert non-string with explicit base")
    
16.         
    
17.         return math.floor(x)
    
18. 
    
19.     if isinstance(x, bytes):
    
20.         if base != 10:
    
21.             raise TypeError("atoi() can't convert non-string with explicit base")
    
22.         
    
23.         x = x.decode()
    
24. 
    
25.     if isinstance(x, str):
    
26.         num = 0
    
27.         current_num = 0
    
28. 
    
29.         if not base:
    
30.             base = 10
    
31. 
    
32.         if x:
    
33.             symbol = -1 if x[0] == "-" else 1
    
34.             x = x[1:] if x[0] in "+-" else x
    
35. 
    
36.             bit = pow(base, len(x)-1)
    
37. 
    
38.         else:
    
39.             raise ValueError('invalid literal for atoi() with base 10: ""')
    
40. 
    
41.         for i in x:
    
42.             if i not in "0123456789ABCDEF":
    
43.                 raise ValueError(f'invalid literal for atoi() with base {base}: "{x}"')
    
44. 
    
45.             elif i == "0":
    
46.                 pass
    
47. 
    
48.             elif i == "1":
    
49.                 num += 1 * bit * symbol
    
50.                 current_num = 1
    
51. 
    
52.             elif i == "2":
    
53.                 num += 2 * bit * symbol
    
54.                 current_num = 2
    
55.                
    
56.             elif i == "3":
    
57.                 num += 3 * bit * symbol
    
58.                 current_num = 3
    
59.                
    
60.             elif i == "4":
    
61.                 num += 4 * bit * symbol
    
62.                 current_num = 4
    
63.                
    
64.             elif i == "5":
    
65.                 num += 5 * bit * symbol
    
66.                 current_num = 5
    
67.                
    
68.             elif i == "6":
    
69.                 num += 6 * bit * symbol
    
70.                 current_num = 6
    
71.                
    
72.             elif i == "7":
    
73.                 num += 7 * bit * symbol
    
74.                 current_num = 7
    
75.                
    
76.             elif i == "8":
    
77.                 num += 8 * bit * symbol
    
78.                 current_num = 8
    
79.                
    
80.             elif i == "9":
    
81.                 num += 9 * bit * symbol
    
82.                 current_num = 9
    
83. 
    
84.             elif i == "A":
    
85.                 num += 10 * bit * symbol
    
86.                 current_num = 10
    
87. 
    
88.             elif i == "B":
    
89.                 num += 11 * bit * symbol
    
90.                 current_num = 11
    
91. 
    
92.             elif i == "C":
    
93.                 num += 12 * bit * symbol
    
94.                 current_num = 12
    
95. 
    
96.             elif i == "D":
    
97.                 num += 13 * bit * symbol
    
98.                 current_num = 13
    
99. 
    
100.             elif i == "E":
     
101.                 num += 14 * bit * symbol
     
102.                 current_num = 14
     
103. 
     
104.             elif i == "F":
     
105.                 num += 15 * bit * symbol
     
106.                 current_num = 15
     
107. 
     
108.             if current_num >= base:
     
109.                 raise ValueError(f'invalid literal for atoi() with base {base}: "{x}"')
     
110. 
     
111.             bit //= base
     
112. 
     
113.         return num
     
114. 
     
115.     raise TypeError(f'atoi() argument must be a string, a bytes-like object or a number, not "{type(x).__name__}"')

复制代码