mypy-dev 1.9.0.dev0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (1018) hide show
  1. mypy/__init__.py +1 -0
  2. mypy/__main__.py +37 -0
  3. mypy/api.py +94 -0
  4. mypy/applytype.py +156 -0
  5. mypy/argmap.py +268 -0
  6. mypy/binder.py +538 -0
  7. mypy/bogus_type.py +27 -0
  8. mypy/build.py +3562 -0
  9. mypy/checker.py +8372 -0
  10. mypy/checkexpr.py +6615 -0
  11. mypy/checkmember.py +1356 -0
  12. mypy/checkpattern.py +801 -0
  13. mypy/checkstrformat.py +1109 -0
  14. mypy/config_parser.py +670 -0
  15. mypy/constant_fold.py +187 -0
  16. mypy/constraints.py +1624 -0
  17. mypy/copytype.py +133 -0
  18. mypy/defaults.py +46 -0
  19. mypy/dmypy/__init__.py +0 -0
  20. mypy/dmypy/__main__.py +6 -0
  21. mypy/dmypy/client.py +749 -0
  22. mypy/dmypy_os.py +42 -0
  23. mypy/dmypy_server.py +1107 -0
  24. mypy/dmypy_util.py +117 -0
  25. mypy/erasetype.py +278 -0
  26. mypy/errorcodes.py +285 -0
  27. mypy/errors.py +1280 -0
  28. mypy/evalexpr.py +205 -0
  29. mypy/expandtype.py +510 -0
  30. mypy/exprtotype.py +206 -0
  31. mypy/fastparse.py +2147 -0
  32. mypy/find_sources.py +243 -0
  33. mypy/fixup.py +426 -0
  34. mypy/freetree.py +23 -0
  35. mypy/fscache.py +309 -0
  36. mypy/fswatcher.py +106 -0
  37. mypy/gclogger.py +47 -0
  38. mypy/git.py +34 -0
  39. mypy/graph_utils.py +112 -0
  40. mypy/indirection.py +121 -0
  41. mypy/infer.py +75 -0
  42. mypy/inspections.py +627 -0
  43. mypy/ipc.py +310 -0
  44. mypy/join.py +866 -0
  45. mypy/literals.py +306 -0
  46. mypy/lookup.py +61 -0
  47. mypy/main.py +1574 -0
  48. mypy/maptype.py +106 -0
  49. mypy/meet.py +1140 -0
  50. mypy/memprofile.py +121 -0
  51. mypy/message_registry.py +326 -0
  52. mypy/messages.py +3171 -0
  53. mypy/metastore.py +225 -0
  54. mypy/mixedtraverser.py +112 -0
  55. mypy/modulefinder.py +875 -0
  56. mypy/moduleinspect.py +184 -0
  57. mypy/mro.py +62 -0
  58. mypy/nodes.py +4112 -0
  59. mypy/operators.py +126 -0
  60. mypy/options.py +556 -0
  61. mypy/parse.py +30 -0
  62. mypy/partially_defined.py +675 -0
  63. mypy/patterns.py +150 -0
  64. mypy/plugin.py +901 -0
  65. mypy/plugins/__init__.py +0 -0
  66. mypy/plugins/attrs.py +1152 -0
  67. mypy/plugins/common.py +440 -0
  68. mypy/plugins/ctypes.py +245 -0
  69. mypy/plugins/dataclasses.py +1101 -0
  70. mypy/plugins/default.py +518 -0
  71. mypy/plugins/enums.py +259 -0
  72. mypy/plugins/functools.py +104 -0
  73. mypy/plugins/proper_plugin.py +175 -0
  74. mypy/plugins/singledispatch.py +224 -0
  75. mypy/py.typed +1 -0
  76. mypy/pyinfo.py +78 -0
  77. mypy/reachability.py +362 -0
  78. mypy/refinfo.py +92 -0
  79. mypy/renaming.py +568 -0
  80. mypy/report.py +924 -0
  81. mypy/scope.py +125 -0
  82. mypy/semanal.py +7023 -0
  83. mypy/semanal_classprop.py +187 -0
  84. mypy/semanal_enum.py +253 -0
  85. mypy/semanal_infer.py +128 -0
  86. mypy/semanal_main.py +511 -0
  87. mypy/semanal_namedtuple.py +670 -0
  88. mypy/semanal_newtype.py +273 -0
  89. mypy/semanal_pass1.py +156 -0
  90. mypy/semanal_shared.py +490 -0
  91. mypy/semanal_typeargs.py +265 -0
  92. mypy/semanal_typeddict.py +575 -0
  93. mypy/server/__init__.py +0 -0
  94. mypy/server/astdiff.py +518 -0
  95. mypy/server/astmerge.py +562 -0
  96. mypy/server/aststrip.py +281 -0
  97. mypy/server/deps.py +1137 -0
  98. mypy/server/mergecheck.py +83 -0
  99. mypy/server/objgraph.py +101 -0
  100. mypy/server/subexpr.py +198 -0
  101. mypy/server/target.py +11 -0
  102. mypy/server/trigger.py +26 -0
  103. mypy/server/update.py +1339 -0
  104. mypy/sharedparse.py +112 -0
  105. mypy/solve.py +562 -0
  106. mypy/split_namespace.py +35 -0
  107. mypy/state.py +28 -0
  108. mypy/stats.py +489 -0
  109. mypy/strconv.py +641 -0
  110. mypy/stubdoc.py +491 -0
  111. mypy/stubgen.py +1873 -0
  112. mypy/stubgenc.py +993 -0
  113. mypy/stubinfo.py +173 -0
  114. mypy/stubtest.py +2079 -0
  115. mypy/stubutil.py +807 -0
  116. mypy/subtypes.py +1967 -0
  117. mypy/suggestions.py +1046 -0
  118. mypy/test/__init__.py +0 -0
  119. mypy/test/config.py +28 -0
  120. mypy/test/data.py +823 -0
  121. mypy/test/helpers.py +476 -0
  122. mypy/test/meta/__init__.py +0 -0
  123. mypy/test/meta/_pytest.py +72 -0
  124. mypy/test/meta/test_diff_helper.py +47 -0
  125. mypy/test/meta/test_parse_data.py +73 -0
  126. mypy/test/meta/test_update_data.py +135 -0
  127. mypy/test/test_find_sources.py +376 -0
  128. mypy/test/test_ref_info.py +45 -0
  129. mypy/test/testapi.py +45 -0
  130. mypy/test/testargs.py +77 -0
  131. mypy/test/testcheck.py +322 -0
  132. mypy/test/testcmdline.py +152 -0
  133. mypy/test/testconstraints.py +134 -0
  134. mypy/test/testdaemon.py +132 -0
  135. mypy/test/testdeps.py +77 -0
  136. mypy/test/testdiff.py +67 -0
  137. mypy/test/testerrorstream.py +46 -0
  138. mypy/test/testfinegrained.py +438 -0
  139. mypy/test/testfinegrainedcache.py +18 -0
  140. mypy/test/testformatter.py +85 -0
  141. mypy/test/testfscache.py +101 -0
  142. mypy/test/testgraph.py +83 -0
  143. mypy/test/testinfer.py +373 -0
  144. mypy/test/testipc.py +119 -0
  145. mypy/test/testmerge.py +238 -0
  146. mypy/test/testmodulefinder.py +278 -0
  147. mypy/test/testmypyc.py +14 -0
  148. mypy/test/testparse.py +107 -0
  149. mypy/test/testpep561.py +211 -0
  150. mypy/test/testpythoneval.py +117 -0
  151. mypy/test/testreports.py +55 -0
  152. mypy/test/testsemanal.py +209 -0
  153. mypy/test/testsolve.py +285 -0
  154. mypy/test/teststubgen.py +1412 -0
  155. mypy/test/teststubinfo.py +12 -0
  156. mypy/test/teststubtest.py +2492 -0
  157. mypy/test/testsubtypes.py +303 -0
  158. mypy/test/testtransform.py +64 -0
  159. mypy/test/testtypegen.py +83 -0
  160. mypy/test/testtypes.py +1551 -0
  161. mypy/test/testutil.py +111 -0
  162. mypy/test/typefixture.py +415 -0
  163. mypy/test/update_data.py +87 -0
  164. mypy/test/visitors.py +63 -0
  165. mypy/traverser.py +961 -0
  166. mypy/treetransform.py +800 -0
  167. mypy/tvar_scope.py +147 -0
  168. mypy/type_visitor.py +564 -0
  169. mypy/typeanal.py +2510 -0
  170. mypy/typeops.py +1082 -0
  171. mypy/types.py +3686 -0
  172. mypy/types_utils.py +166 -0
  173. mypy/typeshed/LICENSE +237 -0
  174. mypy/typeshed/stdlib/VERSIONS +308 -0
  175. mypy/typeshed/stdlib/__future__.pyi +36 -0
  176. mypy/typeshed/stdlib/__main__.pyi +3 -0
  177. mypy/typeshed/stdlib/_ast.pyi +587 -0
  178. mypy/typeshed/stdlib/_bisect.pyi +84 -0
  179. mypy/typeshed/stdlib/_bootlocale.pyi +1 -0
  180. mypy/typeshed/stdlib/_codecs.pyi +133 -0
  181. mypy/typeshed/stdlib/_collections_abc.pyi +94 -0
  182. mypy/typeshed/stdlib/_compat_pickle.pyi +8 -0
  183. mypy/typeshed/stdlib/_compression.pyi +25 -0
  184. mypy/typeshed/stdlib/_csv.pyi +90 -0
  185. mypy/typeshed/stdlib/_ctypes.pyi +205 -0
  186. mypy/typeshed/stdlib/_curses.pyi +557 -0
  187. mypy/typeshed/stdlib/_decimal.pyi +281 -0
  188. mypy/typeshed/stdlib/_dummy_thread.pyi +27 -0
  189. mypy/typeshed/stdlib/_dummy_threading.pyi +164 -0
  190. mypy/typeshed/stdlib/_heapq.pyi +11 -0
  191. mypy/typeshed/stdlib/_imp.pyi +28 -0
  192. mypy/typeshed/stdlib/_json.pyi +49 -0
  193. mypy/typeshed/stdlib/_locale.pyi +100 -0
  194. mypy/typeshed/stdlib/_markupbase.pyi +16 -0
  195. mypy/typeshed/stdlib/_msi.pyi +91 -0
  196. mypy/typeshed/stdlib/_operator.pyi +147 -0
  197. mypy/typeshed/stdlib/_osx_support.pyi +34 -0
  198. mypy/typeshed/stdlib/_posixsubprocess.pyi +32 -0
  199. mypy/typeshed/stdlib/_py_abc.pyi +14 -0
  200. mypy/typeshed/stdlib/_pydecimal.pyi +43 -0
  201. mypy/typeshed/stdlib/_random.pyi +12 -0
  202. mypy/typeshed/stdlib/_sitebuiltins.pyi +16 -0
  203. mypy/typeshed/stdlib/_socket.pyi +802 -0
  204. mypy/typeshed/stdlib/_stat.pyi +103 -0
  205. mypy/typeshed/stdlib/_thread.pyi +52 -0
  206. mypy/typeshed/stdlib/_threading_local.pyi +17 -0
  207. mypy/typeshed/stdlib/_tkinter.pyi +120 -0
  208. mypy/typeshed/stdlib/_tracemalloc.pyi +17 -0
  209. mypy/typeshed/stdlib/_typeshed/__init__.pyi +347 -0
  210. mypy/typeshed/stdlib/_typeshed/dbapi.pyi +37 -0
  211. mypy/typeshed/stdlib/_typeshed/wsgi.pyi +44 -0
  212. mypy/typeshed/stdlib/_typeshed/xml.pyi +9 -0
  213. mypy/typeshed/stdlib/_warnings.pyi +55 -0
  214. mypy/typeshed/stdlib/_weakref.pyi +41 -0
  215. mypy/typeshed/stdlib/_weakrefset.pyi +51 -0
  216. mypy/typeshed/stdlib/_winapi.pyi +251 -0
  217. mypy/typeshed/stdlib/abc.pyi +49 -0
  218. mypy/typeshed/stdlib/aifc.pyi +91 -0
  219. mypy/typeshed/stdlib/antigravity.pyi +3 -0
  220. mypy/typeshed/stdlib/argparse.pyi +571 -0
  221. mypy/typeshed/stdlib/array.pyi +92 -0
  222. mypy/typeshed/stdlib/ast.pyi +277 -0
  223. mypy/typeshed/stdlib/asynchat.pyi +21 -0
  224. mypy/typeshed/stdlib/asyncio/__init__.pyi +41 -0
  225. mypy/typeshed/stdlib/asyncio/base_events.pyi +440 -0
  226. mypy/typeshed/stdlib/asyncio/base_futures.pyi +19 -0
  227. mypy/typeshed/stdlib/asyncio/base_subprocess.pyi +63 -0
  228. mypy/typeshed/stdlib/asyncio/base_tasks.pyi +9 -0
  229. mypy/typeshed/stdlib/asyncio/constants.pyi +20 -0
  230. mypy/typeshed/stdlib/asyncio/coroutines.pyi +26 -0
  231. mypy/typeshed/stdlib/asyncio/events.pyi +574 -0
  232. mypy/typeshed/stdlib/asyncio/exceptions.pyi +43 -0
  233. mypy/typeshed/stdlib/asyncio/format_helpers.pyi +20 -0
  234. mypy/typeshed/stdlib/asyncio/futures.pyi +57 -0
  235. mypy/typeshed/stdlib/asyncio/locks.pyi +121 -0
  236. mypy/typeshed/stdlib/asyncio/log.pyi +3 -0
  237. mypy/typeshed/stdlib/asyncio/mixins.pyi +9 -0
  238. mypy/typeshed/stdlib/asyncio/proactor_events.pyi +64 -0
  239. mypy/typeshed/stdlib/asyncio/protocols.pyi +34 -0
  240. mypy/typeshed/stdlib/asyncio/queues.pyi +47 -0
  241. mypy/typeshed/stdlib/asyncio/runners.pyi +32 -0
  242. mypy/typeshed/stdlib/asyncio/selector_events.pyi +8 -0
  243. mypy/typeshed/stdlib/asyncio/sslproto.pyi +163 -0
  244. mypy/typeshed/stdlib/asyncio/staggered.pyi +10 -0
  245. mypy/typeshed/stdlib/asyncio/streams.pyi +148 -0
  246. mypy/typeshed/stdlib/asyncio/subprocess.pyi +229 -0
  247. mypy/typeshed/stdlib/asyncio/taskgroups.pyi +25 -0
  248. mypy/typeshed/stdlib/asyncio/tasks.pyi +477 -0
  249. mypy/typeshed/stdlib/asyncio/threads.pyi +9 -0
  250. mypy/typeshed/stdlib/asyncio/timeouts.pyi +19 -0
  251. mypy/typeshed/stdlib/asyncio/transports.pyi +47 -0
  252. mypy/typeshed/stdlib/asyncio/trsock.pyi +98 -0
  253. mypy/typeshed/stdlib/asyncio/unix_events.pyi +173 -0
  254. mypy/typeshed/stdlib/asyncio/windows_events.pyi +83 -0
  255. mypy/typeshed/stdlib/asyncio/windows_utils.pyi +49 -0
  256. mypy/typeshed/stdlib/asyncore.pyi +90 -0
  257. mypy/typeshed/stdlib/atexit.pyi +12 -0
  258. mypy/typeshed/stdlib/audioop.pyi +42 -0
  259. mypy/typeshed/stdlib/base64.pyi +59 -0
  260. mypy/typeshed/stdlib/bdb.pyi +102 -0
  261. mypy/typeshed/stdlib/binascii.pyi +36 -0
  262. mypy/typeshed/stdlib/binhex.pyi +45 -0
  263. mypy/typeshed/stdlib/bisect.pyi +4 -0
  264. mypy/typeshed/stdlib/builtins.pyi +1937 -0
  265. mypy/typeshed/stdlib/bz2.pyi +146 -0
  266. mypy/typeshed/stdlib/cProfile.pyi +35 -0
  267. mypy/typeshed/stdlib/calendar.pyi +206 -0
  268. mypy/typeshed/stdlib/cgi.pyi +118 -0
  269. mypy/typeshed/stdlib/cgitb.pyi +32 -0
  270. mypy/typeshed/stdlib/chunk.pyi +20 -0
  271. mypy/typeshed/stdlib/cmath.pyi +36 -0
  272. mypy/typeshed/stdlib/cmd.pyi +45 -0
  273. mypy/typeshed/stdlib/code.pyi +33 -0
  274. mypy/typeshed/stdlib/codecs.pyi +285 -0
  275. mypy/typeshed/stdlib/codeop.pyi +13 -0
  276. mypy/typeshed/stdlib/collections/__init__.pyi +477 -0
  277. mypy/typeshed/stdlib/collections/abc.pyi +2 -0
  278. mypy/typeshed/stdlib/colorsys.pyi +13 -0
  279. mypy/typeshed/stdlib/compileall.pyi +111 -0
  280. mypy/typeshed/stdlib/concurrent/__init__.pyi +0 -0
  281. mypy/typeshed/stdlib/concurrent/futures/__init__.pyi +32 -0
  282. mypy/typeshed/stdlib/concurrent/futures/_base.pyi +120 -0
  283. mypy/typeshed/stdlib/concurrent/futures/process.pyi +233 -0
  284. mypy/typeshed/stdlib/concurrent/futures/thread.pyi +80 -0
  285. mypy/typeshed/stdlib/configparser.pyi +313 -0
  286. mypy/typeshed/stdlib/contextlib.pyi +207 -0
  287. mypy/typeshed/stdlib/contextvars.pyi +63 -0
  288. mypy/typeshed/stdlib/copy.pyi +16 -0
  289. mypy/typeshed/stdlib/copyreg.pyi +21 -0
  290. mypy/typeshed/stdlib/crypt.pyi +12 -0
  291. mypy/typeshed/stdlib/csv.pyi +147 -0
  292. mypy/typeshed/stdlib/ctypes/__init__.pyi +187 -0
  293. mypy/typeshed/stdlib/ctypes/_endian.pyi +19 -0
  294. mypy/typeshed/stdlib/ctypes/util.pyi +6 -0
  295. mypy/typeshed/stdlib/ctypes/wintypes.pyi +298 -0
  296. mypy/typeshed/stdlib/curses/__init__.pyi +21 -0
  297. mypy/typeshed/stdlib/curses/ascii.pyi +63 -0
  298. mypy/typeshed/stdlib/curses/has_key.pyi +4 -0
  299. mypy/typeshed/stdlib/curses/panel.pyi +25 -0
  300. mypy/typeshed/stdlib/curses/textpad.pyi +13 -0
  301. mypy/typeshed/stdlib/dataclasses.pyi +313 -0
  302. mypy/typeshed/stdlib/datetime.pyi +295 -0
  303. mypy/typeshed/stdlib/dbm/__init__.pyi +95 -0
  304. mypy/typeshed/stdlib/dbm/dumb.pyi +31 -0
  305. mypy/typeshed/stdlib/dbm/gnu.pyi +40 -0
  306. mypy/typeshed/stdlib/dbm/ndbm.pyi +36 -0
  307. mypy/typeshed/stdlib/decimal.pyi +2 -0
  308. mypy/typeshed/stdlib/difflib.pyi +140 -0
  309. mypy/typeshed/stdlib/dis.pyi +144 -0
  310. mypy/typeshed/stdlib/distutils/__init__.pyi +5 -0
  311. mypy/typeshed/stdlib/distutils/archive_util.pyi +20 -0
  312. mypy/typeshed/stdlib/distutils/bcppcompiler.pyi +3 -0
  313. mypy/typeshed/stdlib/distutils/ccompiler.pyi +152 -0
  314. mypy/typeshed/stdlib/distutils/cmd.pyi +66 -0
  315. mypy/typeshed/stdlib/distutils/command/__init__.pyi +0 -0
  316. mypy/typeshed/stdlib/distutils/command/bdist.pyi +25 -0
  317. mypy/typeshed/stdlib/distutils/command/bdist_dumb.pyi +21 -0
  318. mypy/typeshed/stdlib/distutils/command/bdist_msi.pyi +45 -0
  319. mypy/typeshed/stdlib/distutils/command/bdist_packager.pyi +0 -0
  320. mypy/typeshed/stdlib/distutils/command/bdist_rpm.pyi +52 -0
  321. mypy/typeshed/stdlib/distutils/command/bdist_wininst.pyi +16 -0
  322. mypy/typeshed/stdlib/distutils/command/build.pyi +31 -0
  323. mypy/typeshed/stdlib/distutils/command/build_clib.pyi +27 -0
  324. mypy/typeshed/stdlib/distutils/command/build_ext.pyi +50 -0
  325. mypy/typeshed/stdlib/distutils/command/build_py.pyi +44 -0
  326. mypy/typeshed/stdlib/distutils/command/build_scripts.pyi +24 -0
  327. mypy/typeshed/stdlib/distutils/command/check.pyi +39 -0
  328. mypy/typeshed/stdlib/distutils/command/clean.pyi +17 -0
  329. mypy/typeshed/stdlib/distutils/command/config.pyi +83 -0
  330. mypy/typeshed/stdlib/distutils/command/install.pyi +63 -0
  331. mypy/typeshed/stdlib/distutils/command/install_data.pyi +19 -0
  332. mypy/typeshed/stdlib/distutils/command/install_egg_info.pyi +18 -0
  333. mypy/typeshed/stdlib/distutils/command/install_headers.pyi +16 -0
  334. mypy/typeshed/stdlib/distutils/command/install_lib.pyi +25 -0
  335. mypy/typeshed/stdlib/distutils/command/install_scripts.pyi +18 -0
  336. mypy/typeshed/stdlib/distutils/command/register.pyi +18 -0
  337. mypy/typeshed/stdlib/distutils/command/sdist.pyi +42 -0
  338. mypy/typeshed/stdlib/distutils/command/upload.pyi +17 -0
  339. mypy/typeshed/stdlib/distutils/config.pyi +17 -0
  340. mypy/typeshed/stdlib/distutils/core.pyi +57 -0
  341. mypy/typeshed/stdlib/distutils/cygwinccompiler.pyi +20 -0
  342. mypy/typeshed/stdlib/distutils/debug.pyi +1 -0
  343. mypy/typeshed/stdlib/distutils/dep_util.pyi +3 -0
  344. mypy/typeshed/stdlib/distutils/dir_util.pyi +13 -0
  345. mypy/typeshed/stdlib/distutils/dist.pyi +146 -0
  346. mypy/typeshed/stdlib/distutils/errors.pyi +19 -0
  347. mypy/typeshed/stdlib/distutils/extension.pyi +36 -0
  348. mypy/typeshed/stdlib/distutils/fancy_getopt.pyi +34 -0
  349. mypy/typeshed/stdlib/distutils/file_util.pyi +14 -0
  350. mypy/typeshed/stdlib/distutils/filelist.pyi +50 -0
  351. mypy/typeshed/stdlib/distutils/log.pyi +25 -0
  352. mypy/typeshed/stdlib/distutils/msvccompiler.pyi +3 -0
  353. mypy/typeshed/stdlib/distutils/spawn.pyi +2 -0
  354. mypy/typeshed/stdlib/distutils/sysconfig.pyi +22 -0
  355. mypy/typeshed/stdlib/distutils/text_file.pyi +21 -0
  356. mypy/typeshed/stdlib/distutils/unixccompiler.pyi +3 -0
  357. mypy/typeshed/stdlib/distutils/util.pyi +46 -0
  358. mypy/typeshed/stdlib/distutils/version.pyi +36 -0
  359. mypy/typeshed/stdlib/doctest.pyi +248 -0
  360. mypy/typeshed/stdlib/dummy_threading.pyi +2 -0
  361. mypy/typeshed/stdlib/email/__init__.pyi +29 -0
  362. mypy/typeshed/stdlib/email/_header_value_parser.pyi +392 -0
  363. mypy/typeshed/stdlib/email/_policybase.pyi +51 -0
  364. mypy/typeshed/stdlib/email/base64mime.pyi +13 -0
  365. mypy/typeshed/stdlib/email/charset.pyi +34 -0
  366. mypy/typeshed/stdlib/email/contentmanager.pyi +11 -0
  367. mypy/typeshed/stdlib/email/encoders.pyi +8 -0
  368. mypy/typeshed/stdlib/email/errors.pyi +39 -0
  369. mypy/typeshed/stdlib/email/feedparser.pyi +23 -0
  370. mypy/typeshed/stdlib/email/generator.pyi +40 -0
  371. mypy/typeshed/stdlib/email/header.pyi +31 -0
  372. mypy/typeshed/stdlib/email/headerregistry.pyi +178 -0
  373. mypy/typeshed/stdlib/email/iterators.pyi +12 -0
  374. mypy/typeshed/stdlib/email/message.pyi +161 -0
  375. mypy/typeshed/stdlib/email/mime/__init__.pyi +0 -0
  376. mypy/typeshed/stdlib/email/mime/application.pyi +17 -0
  377. mypy/typeshed/stdlib/email/mime/audio.pyi +17 -0
  378. mypy/typeshed/stdlib/email/mime/base.pyi +8 -0
  379. mypy/typeshed/stdlib/email/mime/image.pyi +17 -0
  380. mypy/typeshed/stdlib/email/mime/message.pyi +8 -0
  381. mypy/typeshed/stdlib/email/mime/multipart.pyi +18 -0
  382. mypy/typeshed/stdlib/email/mime/nonmultipart.pyi +5 -0
  383. mypy/typeshed/stdlib/email/mime/text.pyi +9 -0
  384. mypy/typeshed/stdlib/email/parser.pyi +26 -0
  385. mypy/typeshed/stdlib/email/policy.pyi +38 -0
  386. mypy/typeshed/stdlib/email/quoprimime.pyi +28 -0
  387. mypy/typeshed/stdlib/email/utils.pyi +66 -0
  388. mypy/typeshed/stdlib/encodings/__init__.pyi +10 -0
  389. mypy/typeshed/stdlib/encodings/utf_8.pyi +21 -0
  390. mypy/typeshed/stdlib/encodings/utf_8_sig.pyi +22 -0
  391. mypy/typeshed/stdlib/ensurepip/__init__.pyi +12 -0
  392. mypy/typeshed/stdlib/enum.pyi +317 -0
  393. mypy/typeshed/stdlib/errno.pyi +222 -0
  394. mypy/typeshed/stdlib/faulthandler.pyi +13 -0
  395. mypy/typeshed/stdlib/fcntl.pyi +127 -0
  396. mypy/typeshed/stdlib/filecmp.pyi +57 -0
  397. mypy/typeshed/stdlib/fileinput.pyi +213 -0
  398. mypy/typeshed/stdlib/fnmatch.pyi +9 -0
  399. mypy/typeshed/stdlib/formatter.pyi +88 -0
  400. mypy/typeshed/stdlib/fractions.pyi +150 -0
  401. mypy/typeshed/stdlib/ftplib.pyi +178 -0
  402. mypy/typeshed/stdlib/functools.pyi +200 -0
  403. mypy/typeshed/stdlib/gc.pyi +37 -0
  404. mypy/typeshed/stdlib/genericpath.pyi +52 -0
  405. mypy/typeshed/stdlib/getopt.pyi +11 -0
  406. mypy/typeshed/stdlib/getpass.pyi +8 -0
  407. mypy/typeshed/stdlib/gettext.pyi +169 -0
  408. mypy/typeshed/stdlib/glob.pyi +42 -0
  409. mypy/typeshed/stdlib/graphlib.pyi +28 -0
  410. mypy/typeshed/stdlib/grp.pyi +21 -0
  411. mypy/typeshed/stdlib/gzip.pyi +160 -0
  412. mypy/typeshed/stdlib/hashlib.pyi +165 -0
  413. mypy/typeshed/stdlib/heapq.pyi +17 -0
  414. mypy/typeshed/stdlib/hmac.pyi +38 -0
  415. mypy/typeshed/stdlib/html/__init__.pyi +6 -0
  416. mypy/typeshed/stdlib/html/entities.pyi +6 -0
  417. mypy/typeshed/stdlib/html/parser.pyi +34 -0
  418. mypy/typeshed/stdlib/http/__init__.pyi +105 -0
  419. mypy/typeshed/stdlib/http/client.pyi +259 -0
  420. mypy/typeshed/stdlib/http/cookiejar.pyi +159 -0
  421. mypy/typeshed/stdlib/http/cookies.pyi +60 -0
  422. mypy/typeshed/stdlib/http/server.pyi +83 -0
  423. mypy/typeshed/stdlib/imaplib.pyi +168 -0
  424. mypy/typeshed/stdlib/imghdr.pyi +17 -0
  425. mypy/typeshed/stdlib/imp.pyi +62 -0
  426. mypy/typeshed/stdlib/importlib/__init__.pyi +24 -0
  427. mypy/typeshed/stdlib/importlib/_abc.pyi +15 -0
  428. mypy/typeshed/stdlib/importlib/abc.pyi +172 -0
  429. mypy/typeshed/stdlib/importlib/machinery.pyi +179 -0
  430. mypy/typeshed/stdlib/importlib/metadata/__init__.pyi +276 -0
  431. mypy/typeshed/stdlib/importlib/metadata/_meta.pyi +40 -0
  432. mypy/typeshed/stdlib/importlib/readers.pyi +68 -0
  433. mypy/typeshed/stdlib/importlib/resources/__init__.pyi +46 -0
  434. mypy/typeshed/stdlib/importlib/resources/abc.pyi +12 -0
  435. mypy/typeshed/stdlib/importlib/resources/readers.pyi +14 -0
  436. mypy/typeshed/stdlib/importlib/resources/simple.pyi +49 -0
  437. mypy/typeshed/stdlib/importlib/simple.pyi +11 -0
  438. mypy/typeshed/stdlib/importlib/util.pyi +43 -0
  439. mypy/typeshed/stdlib/inspect.pyi +631 -0
  440. mypy/typeshed/stdlib/io.pyi +204 -0
  441. mypy/typeshed/stdlib/ipaddress.pyi +208 -0
  442. mypy/typeshed/stdlib/itertools.pyi +293 -0
  443. mypy/typeshed/stdlib/json/__init__.pyi +61 -0
  444. mypy/typeshed/stdlib/json/decoder.pyi +32 -0
  445. mypy/typeshed/stdlib/json/encoder.pyi +38 -0
  446. mypy/typeshed/stdlib/json/tool.pyi +1 -0
  447. mypy/typeshed/stdlib/keyword.pyi +21 -0
  448. mypy/typeshed/stdlib/lib2to3/__init__.pyi +0 -0
  449. mypy/typeshed/stdlib/lib2to3/btm_matcher.pyi +28 -0
  450. mypy/typeshed/stdlib/lib2to3/fixer_base.pyi +42 -0
  451. mypy/typeshed/stdlib/lib2to3/fixes/__init__.pyi +0 -0
  452. mypy/typeshed/stdlib/lib2to3/fixes/fix_apply.pyi +8 -0
  453. mypy/typeshed/stdlib/lib2to3/fixes/fix_asserts.pyi +10 -0
  454. mypy/typeshed/stdlib/lib2to3/fixes/fix_basestring.pyi +8 -0
  455. mypy/typeshed/stdlib/lib2to3/fixes/fix_buffer.pyi +8 -0
  456. mypy/typeshed/stdlib/lib2to3/fixes/fix_dict.pyi +16 -0
  457. mypy/typeshed/stdlib/lib2to3/fixes/fix_except.pyi +14 -0
  458. mypy/typeshed/stdlib/lib2to3/fixes/fix_exec.pyi +8 -0
  459. mypy/typeshed/stdlib/lib2to3/fixes/fix_execfile.pyi +8 -0
  460. mypy/typeshed/stdlib/lib2to3/fixes/fix_exitfunc.pyi +13 -0
  461. mypy/typeshed/stdlib/lib2to3/fixes/fix_filter.pyi +9 -0
  462. mypy/typeshed/stdlib/lib2to3/fixes/fix_funcattrs.pyi +8 -0
  463. mypy/typeshed/stdlib/lib2to3/fixes/fix_future.pyi +8 -0
  464. mypy/typeshed/stdlib/lib2to3/fixes/fix_getcwdu.pyi +8 -0
  465. mypy/typeshed/stdlib/lib2to3/fixes/fix_has_key.pyi +8 -0
  466. mypy/typeshed/stdlib/lib2to3/fixes/fix_idioms.pyi +15 -0
  467. mypy/typeshed/stdlib/lib2to3/fixes/fix_import.pyi +16 -0
  468. mypy/typeshed/stdlib/lib2to3/fixes/fix_imports.pyi +21 -0
  469. mypy/typeshed/stdlib/lib2to3/fixes/fix_imports2.pyi +6 -0
  470. mypy/typeshed/stdlib/lib2to3/fixes/fix_input.pyi +11 -0
  471. mypy/typeshed/stdlib/lib2to3/fixes/fix_intern.pyi +9 -0
  472. mypy/typeshed/stdlib/lib2to3/fixes/fix_isinstance.pyi +8 -0
  473. mypy/typeshed/stdlib/lib2to3/fixes/fix_itertools.pyi +9 -0
  474. mypy/typeshed/stdlib/lib2to3/fixes/fix_itertools_imports.pyi +7 -0
  475. mypy/typeshed/stdlib/lib2to3/fixes/fix_long.pyi +7 -0
  476. mypy/typeshed/stdlib/lib2to3/fixes/fix_map.pyi +9 -0
  477. mypy/typeshed/stdlib/lib2to3/fixes/fix_metaclass.pyi +17 -0
  478. mypy/typeshed/stdlib/lib2to3/fixes/fix_methodattrs.pyi +10 -0
  479. mypy/typeshed/stdlib/lib2to3/fixes/fix_ne.pyi +8 -0
  480. mypy/typeshed/stdlib/lib2to3/fixes/fix_next.pyi +19 -0
  481. mypy/typeshed/stdlib/lib2to3/fixes/fix_nonzero.pyi +8 -0
  482. mypy/typeshed/stdlib/lib2to3/fixes/fix_numliterals.pyi +8 -0
  483. mypy/typeshed/stdlib/lib2to3/fixes/fix_operator.pyi +12 -0
  484. mypy/typeshed/stdlib/lib2to3/fixes/fix_paren.pyi +8 -0
  485. mypy/typeshed/stdlib/lib2to3/fixes/fix_print.pyi +12 -0
  486. mypy/typeshed/stdlib/lib2to3/fixes/fix_raise.pyi +8 -0
  487. mypy/typeshed/stdlib/lib2to3/fixes/fix_raw_input.pyi +8 -0
  488. mypy/typeshed/stdlib/lib2to3/fixes/fix_reduce.pyi +8 -0
  489. mypy/typeshed/stdlib/lib2to3/fixes/fix_reload.pyi +9 -0
  490. mypy/typeshed/stdlib/lib2to3/fixes/fix_renames.pyi +17 -0
  491. mypy/typeshed/stdlib/lib2to3/fixes/fix_repr.pyi +8 -0
  492. mypy/typeshed/stdlib/lib2to3/fixes/fix_set_literal.pyi +7 -0
  493. mypy/typeshed/stdlib/lib2to3/fixes/fix_standarderror.pyi +8 -0
  494. mypy/typeshed/stdlib/lib2to3/fixes/fix_sys_exc.pyi +9 -0
  495. mypy/typeshed/stdlib/lib2to3/fixes/fix_throw.pyi +8 -0
  496. mypy/typeshed/stdlib/lib2to3/fixes/fix_tuple_params.pyi +17 -0
  497. mypy/typeshed/stdlib/lib2to3/fixes/fix_types.pyi +8 -0
  498. mypy/typeshed/stdlib/lib2to3/fixes/fix_unicode.pyi +12 -0
  499. mypy/typeshed/stdlib/lib2to3/fixes/fix_urllib.pyi +15 -0
  500. mypy/typeshed/stdlib/lib2to3/fixes/fix_ws_comma.pyi +12 -0
  501. mypy/typeshed/stdlib/lib2to3/fixes/fix_xrange.pyi +20 -0
  502. mypy/typeshed/stdlib/lib2to3/fixes/fix_xreadlines.pyi +8 -0
  503. mypy/typeshed/stdlib/lib2to3/fixes/fix_zip.pyi +9 -0
  504. mypy/typeshed/stdlib/lib2to3/main.pyi +42 -0
  505. mypy/typeshed/stdlib/lib2to3/pgen2/__init__.pyi +9 -0
  506. mypy/typeshed/stdlib/lib2to3/pgen2/driver.pyi +27 -0
  507. mypy/typeshed/stdlib/lib2to3/pgen2/grammar.pyi +24 -0
  508. mypy/typeshed/stdlib/lib2to3/pgen2/literals.pyi +7 -0
  509. mypy/typeshed/stdlib/lib2to3/pgen2/parse.pyi +30 -0
  510. mypy/typeshed/stdlib/lib2to3/pgen2/pgen.pyi +50 -0
  511. mypy/typeshed/stdlib/lib2to3/pgen2/token.pyi +67 -0
  512. mypy/typeshed/stdlib/lib2to3/pgen2/tokenize.pyi +96 -0
  513. mypy/typeshed/stdlib/lib2to3/pygram.pyi +114 -0
  514. mypy/typeshed/stdlib/lib2to3/pytree.pyi +117 -0
  515. mypy/typeshed/stdlib/lib2to3/refactor.pyi +82 -0
  516. mypy/typeshed/stdlib/linecache.pyi +23 -0
  517. mypy/typeshed/stdlib/locale.pyi +152 -0
  518. mypy/typeshed/stdlib/logging/__init__.pyi +655 -0
  519. mypy/typeshed/stdlib/logging/config.pyi +134 -0
  520. mypy/typeshed/stdlib/logging/handlers.pyi +275 -0
  521. mypy/typeshed/stdlib/lzma.pyi +197 -0
  522. mypy/typeshed/stdlib/mailbox.pyi +256 -0
  523. mypy/typeshed/stdlib/mailcap.pyi +11 -0
  524. mypy/typeshed/stdlib/marshal.pyi +33 -0
  525. mypy/typeshed/stdlib/math.pyi +125 -0
  526. mypy/typeshed/stdlib/mimetypes.pyi +46 -0
  527. mypy/typeshed/stdlib/mmap.pyi +113 -0
  528. mypy/typeshed/stdlib/modulefinder.pyi +66 -0
  529. mypy/typeshed/stdlib/msilib/__init__.pyi +176 -0
  530. mypy/typeshed/stdlib/msilib/schema.pyi +94 -0
  531. mypy/typeshed/stdlib/msilib/sequence.pyi +13 -0
  532. mypy/typeshed/stdlib/msilib/text.pyi +7 -0
  533. mypy/typeshed/stdlib/msvcrt.pyi +32 -0
  534. mypy/typeshed/stdlib/multiprocessing/__init__.pyi +90 -0
  535. mypy/typeshed/stdlib/multiprocessing/connection.pyi +75 -0
  536. mypy/typeshed/stdlib/multiprocessing/context.pyi +187 -0
  537. mypy/typeshed/stdlib/multiprocessing/dummy/__init__.pyi +77 -0
  538. mypy/typeshed/stdlib/multiprocessing/dummy/connection.pyi +39 -0
  539. mypy/typeshed/stdlib/multiprocessing/forkserver.pyi +31 -0
  540. mypy/typeshed/stdlib/multiprocessing/heap.pyi +36 -0
  541. mypy/typeshed/stdlib/multiprocessing/managers.pyi +212 -0
  542. mypy/typeshed/stdlib/multiprocessing/pool.pyi +103 -0
  543. mypy/typeshed/stdlib/multiprocessing/popen_fork.pyi +23 -0
  544. mypy/typeshed/stdlib/multiprocessing/popen_forkserver.pyi +16 -0
  545. mypy/typeshed/stdlib/multiprocessing/popen_spawn_posix.pyi +20 -0
  546. mypy/typeshed/stdlib/multiprocessing/popen_spawn_win32.pyi +30 -0
  547. mypy/typeshed/stdlib/multiprocessing/process.pyi +39 -0
  548. mypy/typeshed/stdlib/multiprocessing/queues.pyi +41 -0
  549. mypy/typeshed/stdlib/multiprocessing/reduction.pyi +89 -0
  550. mypy/typeshed/stdlib/multiprocessing/resource_sharer.pyi +20 -0
  551. mypy/typeshed/stdlib/multiprocessing/resource_tracker.pyi +18 -0
  552. mypy/typeshed/stdlib/multiprocessing/shared_memory.pyi +40 -0
  553. mypy/typeshed/stdlib/multiprocessing/sharedctypes.pyi +101 -0
  554. mypy/typeshed/stdlib/multiprocessing/spawn.pyi +32 -0
  555. mypy/typeshed/stdlib/multiprocessing/synchronize.pyi +54 -0
  556. mypy/typeshed/stdlib/multiprocessing/util.pyi +79 -0
  557. mypy/typeshed/stdlib/netrc.pyi +23 -0
  558. mypy/typeshed/stdlib/nis.pyi +9 -0
  559. mypy/typeshed/stdlib/nntplib.pyi +125 -0
  560. mypy/typeshed/stdlib/nt.pyi +111 -0
  561. mypy/typeshed/stdlib/ntpath.pyi +119 -0
  562. mypy/typeshed/stdlib/nturl2path.pyi +2 -0
  563. mypy/typeshed/stdlib/numbers.pyi +154 -0
  564. mypy/typeshed/stdlib/opcode.pyi +59 -0
  565. mypy/typeshed/stdlib/operator.pyi +110 -0
  566. mypy/typeshed/stdlib/optparse.pyi +252 -0
  567. mypy/typeshed/stdlib/os/__init__.pyi +1139 -0
  568. mypy/typeshed/stdlib/os/path.pyi +8 -0
  569. mypy/typeshed/stdlib/ossaudiodev.pyi +131 -0
  570. mypy/typeshed/stdlib/parser.pyi +24 -0
  571. mypy/typeshed/stdlib/pathlib.pyi +232 -0
  572. mypy/typeshed/stdlib/pdb.pyi +181 -0
  573. mypy/typeshed/stdlib/pickle.pyi +270 -0
  574. mypy/typeshed/stdlib/pickletools.pyi +167 -0
  575. mypy/typeshed/stdlib/pipes.pyi +16 -0
  576. mypy/typeshed/stdlib/pkgutil.pyi +53 -0
  577. mypy/typeshed/stdlib/platform.pyi +42 -0
  578. mypy/typeshed/stdlib/plistlib.pyi +113 -0
  579. mypy/typeshed/stdlib/poplib.pyi +71 -0
  580. mypy/typeshed/stdlib/posix.pyi +359 -0
  581. mypy/typeshed/stdlib/posixpath.pyi +161 -0
  582. mypy/typeshed/stdlib/pprint.pyi +112 -0
  583. mypy/typeshed/stdlib/profile.pyi +31 -0
  584. mypy/typeshed/stdlib/pstats.pyi +78 -0
  585. mypy/typeshed/stdlib/pty.pyi +19 -0
  586. mypy/typeshed/stdlib/pwd.pyi +27 -0
  587. mypy/typeshed/stdlib/py_compile.pyi +34 -0
  588. mypy/typeshed/stdlib/pyclbr.pyi +74 -0
  589. mypy/typeshed/stdlib/pydoc.pyi +261 -0
  590. mypy/typeshed/stdlib/pydoc_data/__init__.pyi +0 -0
  591. mypy/typeshed/stdlib/pydoc_data/topics.pyi +1 -0
  592. mypy/typeshed/stdlib/pyexpat/__init__.pyi +80 -0
  593. mypy/typeshed/stdlib/pyexpat/errors.pyi +49 -0
  594. mypy/typeshed/stdlib/pyexpat/model.pyi +11 -0
  595. mypy/typeshed/stdlib/queue.pyi +58 -0
  596. mypy/typeshed/stdlib/quopri.pyi +11 -0
  597. mypy/typeshed/stdlib/random.pyi +138 -0
  598. mypy/typeshed/stdlib/re.pyi +290 -0
  599. mypy/typeshed/stdlib/readline.pyi +36 -0
  600. mypy/typeshed/stdlib/reprlib.pyi +65 -0
  601. mypy/typeshed/stdlib/resource.pyi +92 -0
  602. mypy/typeshed/stdlib/rlcompleter.pyi +9 -0
  603. mypy/typeshed/stdlib/runpy.pyi +24 -0
  604. mypy/typeshed/stdlib/sched.pyi +42 -0
  605. mypy/typeshed/stdlib/secrets.pyi +15 -0
  606. mypy/typeshed/stdlib/select.pyi +151 -0
  607. mypy/typeshed/stdlib/selectors.pyi +67 -0
  608. mypy/typeshed/stdlib/shelve.pyi +46 -0
  609. mypy/typeshed/stdlib/shlex.pyi +43 -0
  610. mypy/typeshed/stdlib/shutil.pyi +189 -0
  611. mypy/typeshed/stdlib/signal.pyi +189 -0
  612. mypy/typeshed/stdlib/site.pyi +27 -0
  613. mypy/typeshed/stdlib/smtpd.pyi +91 -0
  614. mypy/typeshed/stdlib/smtplib.pyi +204 -0
  615. mypy/typeshed/stdlib/sndhdr.pyi +14 -0
  616. mypy/typeshed/stdlib/socket.pyi +825 -0
  617. mypy/typeshed/stdlib/socketserver.pyi +168 -0
  618. mypy/typeshed/stdlib/spwd.pyi +40 -0
  619. mypy/typeshed/stdlib/sqlite3/__init__.pyi +1 -0
  620. mypy/typeshed/stdlib/sqlite3/dbapi2.pyi +469 -0
  621. mypy/typeshed/stdlib/sre_compile.pyi +11 -0
  622. mypy/typeshed/stdlib/sre_constants.pyi +130 -0
  623. mypy/typeshed/stdlib/sre_parse.pyi +104 -0
  624. mypy/typeshed/stdlib/ssl.pyi +529 -0
  625. mypy/typeshed/stdlib/stat.pyi +1 -0
  626. mypy/typeshed/stdlib/statistics.pyi +132 -0
  627. mypy/typeshed/stdlib/string.pyi +83 -0
  628. mypy/typeshed/stdlib/stringprep.pyi +27 -0
  629. mypy/typeshed/stdlib/struct.pyi +26 -0
  630. mypy/typeshed/stdlib/subprocess.pyi +2614 -0
  631. mypy/typeshed/stdlib/sunau.pyi +86 -0
  632. mypy/typeshed/stdlib/symbol.pyi +93 -0
  633. mypy/typeshed/stdlib/symtable.pyi +58 -0
  634. mypy/typeshed/stdlib/sys/__init__.pyi +373 -0
  635. mypy/typeshed/stdlib/sys/_monitoring.pyi +52 -0
  636. mypy/typeshed/stdlib/sysconfig.pyi +43 -0
  637. mypy/typeshed/stdlib/syslog.pyi +46 -0
  638. mypy/typeshed/stdlib/tabnanny.pyi +16 -0
  639. mypy/typeshed/stdlib/tarfile.pyi +438 -0
  640. mypy/typeshed/stdlib/telnetlib.pyi +122 -0
  641. mypy/typeshed/stdlib/tempfile.pyi +477 -0
  642. mypy/typeshed/stdlib/termios.pyi +267 -0
  643. mypy/typeshed/stdlib/textwrap.pyi +103 -0
  644. mypy/typeshed/stdlib/this.pyi +2 -0
  645. mypy/typeshed/stdlib/threading.pyi +187 -0
  646. mypy/typeshed/stdlib/time.pyi +107 -0
  647. mypy/typeshed/stdlib/timeit.pyi +32 -0
  648. mypy/typeshed/stdlib/tkinter/__init__.pyi +3641 -0
  649. mypy/typeshed/stdlib/tkinter/colorchooser.pyi +20 -0
  650. mypy/typeshed/stdlib/tkinter/commondialog.pyi +14 -0
  651. mypy/typeshed/stdlib/tkinter/constants.pyi +80 -0
  652. mypy/typeshed/stdlib/tkinter/dialog.pyi +16 -0
  653. mypy/typeshed/stdlib/tkinter/dnd.pyi +20 -0
  654. mypy/typeshed/stdlib/tkinter/filedialog.pyi +151 -0
  655. mypy/typeshed/stdlib/tkinter/font.pyi +116 -0
  656. mypy/typeshed/stdlib/tkinter/messagebox.pyi +44 -0
  657. mypy/typeshed/stdlib/tkinter/scrolledtext.pyi +10 -0
  658. mypy/typeshed/stdlib/tkinter/simpledialog.pyi +54 -0
  659. mypy/typeshed/stdlib/tkinter/tix.pyi +299 -0
  660. mypy/typeshed/stdlib/tkinter/ttk.pyi +1191 -0
  661. mypy/typeshed/stdlib/token.pyi +159 -0
  662. mypy/typeshed/stdlib/tokenize.pyi +177 -0
  663. mypy/typeshed/stdlib/tomllib.pyi +10 -0
  664. mypy/typeshed/stdlib/trace.pyi +79 -0
  665. mypy/typeshed/stdlib/traceback.pyi +260 -0
  666. mypy/typeshed/stdlib/tracemalloc.pyi +124 -0
  667. mypy/typeshed/stdlib/tty.pyi +30 -0
  668. mypy/typeshed/stdlib/turtle.pyi +713 -0
  669. mypy/typeshed/stdlib/types.pyi +605 -0
  670. mypy/typeshed/stdlib/typing.pyi +968 -0
  671. mypy/typeshed/stdlib/typing_extensions.pyi +503 -0
  672. mypy/typeshed/stdlib/unicodedata.pyi +71 -0
  673. mypy/typeshed/stdlib/unittest/__init__.pyi +67 -0
  674. mypy/typeshed/stdlib/unittest/_log.pyi +27 -0
  675. mypy/typeshed/stdlib/unittest/async_case.pyi +21 -0
  676. mypy/typeshed/stdlib/unittest/case.pyi +342 -0
  677. mypy/typeshed/stdlib/unittest/loader.pyi +51 -0
  678. mypy/typeshed/stdlib/unittest/main.pyi +69 -0
  679. mypy/typeshed/stdlib/unittest/mock.pyi +430 -0
  680. mypy/typeshed/stdlib/unittest/result.pyi +46 -0
  681. mypy/typeshed/stdlib/unittest/runner.pyi +72 -0
  682. mypy/typeshed/stdlib/unittest/signals.pyi +15 -0
  683. mypy/typeshed/stdlib/unittest/suite.pyi +22 -0
  684. mypy/typeshed/stdlib/unittest/util.pyi +23 -0
  685. mypy/typeshed/stdlib/urllib/__init__.pyi +0 -0
  686. mypy/typeshed/stdlib/urllib/error.pyi +23 -0
  687. mypy/typeshed/stdlib/urllib/parse.pyi +210 -0
  688. mypy/typeshed/stdlib/urllib/request.pyi +399 -0
  689. mypy/typeshed/stdlib/urllib/response.pyi +43 -0
  690. mypy/typeshed/stdlib/urllib/robotparser.pyi +20 -0
  691. mypy/typeshed/stdlib/uu.pyi +13 -0
  692. mypy/typeshed/stdlib/uuid.pyi +100 -0
  693. mypy/typeshed/stdlib/venv/__init__.pyi +79 -0
  694. mypy/typeshed/stdlib/warnings.pyi +112 -0
  695. mypy/typeshed/stdlib/wave.pyi +85 -0
  696. mypy/typeshed/stdlib/weakref.pyi +150 -0
  697. mypy/typeshed/stdlib/webbrowser.pyi +74 -0
  698. mypy/typeshed/stdlib/winreg.pyi +103 -0
  699. mypy/typeshed/stdlib/winsound.pyi +28 -0
  700. mypy/typeshed/stdlib/wsgiref/__init__.pyi +0 -0
  701. mypy/typeshed/stdlib/wsgiref/handlers.pyi +91 -0
  702. mypy/typeshed/stdlib/wsgiref/headers.pyi +26 -0
  703. mypy/typeshed/stdlib/wsgiref/simple_server.pyi +37 -0
  704. mypy/typeshed/stdlib/wsgiref/types.pyi +32 -0
  705. mypy/typeshed/stdlib/wsgiref/util.pyi +24 -0
  706. mypy/typeshed/stdlib/wsgiref/validate.pyi +50 -0
  707. mypy/typeshed/stdlib/xdrlib.pyi +57 -0
  708. mypy/typeshed/stdlib/xml/__init__.pyi +1 -0
  709. mypy/typeshed/stdlib/xml/dom/NodeFilter.pyi +19 -0
  710. mypy/typeshed/stdlib/xml/dom/__init__.pyi +69 -0
  711. mypy/typeshed/stdlib/xml/dom/domreg.pyi +8 -0
  712. mypy/typeshed/stdlib/xml/dom/expatbuilder.pyi +100 -0
  713. mypy/typeshed/stdlib/xml/dom/minicompat.pyi +22 -0
  714. mypy/typeshed/stdlib/xml/dom/minidom.pyi +404 -0
  715. mypy/typeshed/stdlib/xml/dom/pulldom.pyi +94 -0
  716. mypy/typeshed/stdlib/xml/dom/xmlbuilder.pyi +108 -0
  717. mypy/typeshed/stdlib/xml/etree/ElementInclude.pyi +28 -0
  718. mypy/typeshed/stdlib/xml/etree/ElementPath.pyi +34 -0
  719. mypy/typeshed/stdlib/xml/etree/ElementTree.pyi +321 -0
  720. mypy/typeshed/stdlib/xml/etree/__init__.pyi +0 -0
  721. mypy/typeshed/stdlib/xml/etree/cElementTree.pyi +1 -0
  722. mypy/typeshed/stdlib/xml/parsers/__init__.pyi +1 -0
  723. mypy/typeshed/stdlib/xml/parsers/expat/__init__.pyi +1 -0
  724. mypy/typeshed/stdlib/xml/parsers/expat/errors.pyi +1 -0
  725. mypy/typeshed/stdlib/xml/parsers/expat/model.pyi +1 -0
  726. mypy/typeshed/stdlib/xml/sax/__init__.pyi +25 -0
  727. mypy/typeshed/stdlib/xml/sax/_exceptions.pyi +19 -0
  728. mypy/typeshed/stdlib/xml/sax/handler.pyi +55 -0
  729. mypy/typeshed/stdlib/xml/sax/saxutils.pyi +60 -0
  730. mypy/typeshed/stdlib/xml/sax/xmlreader.pyi +87 -0
  731. mypy/typeshed/stdlib/xmlrpc/__init__.pyi +0 -0
  732. mypy/typeshed/stdlib/xmlrpc/client.pyi +296 -0
  733. mypy/typeshed/stdlib/xmlrpc/server.pyi +143 -0
  734. mypy/typeshed/stdlib/xxlimited.pyi +22 -0
  735. mypy/typeshed/stdlib/zipapp.pyi +20 -0
  736. mypy/typeshed/stdlib/zipfile/__init__.pyi +306 -0
  737. mypy/typeshed/stdlib/zipfile/_path.pyi +95 -0
  738. mypy/typeshed/stdlib/zipimport.pyi +32 -0
  739. mypy/typeshed/stdlib/zlib.pyi +56 -0
  740. mypy/typeshed/stdlib/zoneinfo/__init__.pyi +38 -0
  741. mypy/typeshed/stubs/mypy-extensions/mypy_extensions.pyi +218 -0
  742. mypy/typestate.py +323 -0
  743. mypy/typetraverser.py +142 -0
  744. mypy/typevars.py +93 -0
  745. mypy/typevartuples.py +32 -0
  746. mypy/util.py +869 -0
  747. mypy/version.py +1 -0
  748. mypy/visitor.py +621 -0
  749. mypy/xml/mypy-html.css +104 -0
  750. mypy/xml/mypy-html.xslt +81 -0
  751. mypy/xml/mypy-txt.xslt +100 -0
  752. mypy/xml/mypy.xsd +50 -0
  753. mypy_dev-1.9.0.dev0.dist-info/LICENSE +229 -0
  754. mypy_dev-1.9.0.dev0.dist-info/METADATA +39 -0
  755. mypy_dev-1.9.0.dev0.dist-info/RECORD +1018 -0
  756. mypy_dev-1.9.0.dev0.dist-info/WHEEL +5 -0
  757. mypy_dev-1.9.0.dev0.dist-info/entry_points.txt +6 -0
  758. mypy_dev-1.9.0.dev0.dist-info/top_level.txt +2 -0
  759. mypyc/README.md +133 -0
  760. mypyc/__init__.py +0 -0
  761. mypyc/__main__.py +57 -0
  762. mypyc/analysis/__init__.py +0 -0
  763. mypyc/analysis/attrdefined.py +436 -0
  764. mypyc/analysis/blockfreq.py +32 -0
  765. mypyc/analysis/dataflow.py +624 -0
  766. mypyc/analysis/ircheck.py +429 -0
  767. mypyc/analysis/selfleaks.py +207 -0
  768. mypyc/build.py +616 -0
  769. mypyc/codegen/__init__.py +0 -0
  770. mypyc/codegen/cstring.py +54 -0
  771. mypyc/codegen/emit.py +1193 -0
  772. mypyc/codegen/emitclass.py +1060 -0
  773. mypyc/codegen/emitfunc.py +846 -0
  774. mypyc/codegen/emitmodule.py +1131 -0
  775. mypyc/codegen/emitwrapper.py +979 -0
  776. mypyc/codegen/literals.py +302 -0
  777. mypyc/common.py +136 -0
  778. mypyc/crash.py +31 -0
  779. mypyc/doc/Makefile +20 -0
  780. mypyc/doc/bool_operations.rst +27 -0
  781. mypyc/doc/compilation_units.rst +20 -0
  782. mypyc/doc/conf.py +59 -0
  783. mypyc/doc/cpython-timings.md +25 -0
  784. mypyc/doc/dev-intro.md +548 -0
  785. mypyc/doc/dict_operations.rst +59 -0
  786. mypyc/doc/differences_from_python.rst +332 -0
  787. mypyc/doc/float_operations.rst +50 -0
  788. mypyc/doc/future.md +42 -0
  789. mypyc/doc/getting_started.rst +240 -0
  790. mypyc/doc/index.rst +61 -0
  791. mypyc/doc/int_operations.rst +162 -0
  792. mypyc/doc/introduction.rst +150 -0
  793. mypyc/doc/list_operations.rst +65 -0
  794. mypyc/doc/make.bat +35 -0
  795. mypyc/doc/native_classes.rst +206 -0
  796. mypyc/doc/native_operations.rst +55 -0
  797. mypyc/doc/performance_tips_and_tricks.rst +244 -0
  798. mypyc/doc/set_operations.rst +47 -0
  799. mypyc/doc/str_operations.rst +35 -0
  800. mypyc/doc/tuple_operations.rst +33 -0
  801. mypyc/doc/using_type_annotations.rst +398 -0
  802. mypyc/errors.py +29 -0
  803. mypyc/external/googletest/LICENSE +28 -0
  804. mypyc/external/googletest/README.md +280 -0
  805. mypyc/external/googletest/include/gtest/gtest-death-test.h +294 -0
  806. mypyc/external/googletest/include/gtest/gtest-message.h +250 -0
  807. mypyc/external/googletest/include/gtest/gtest-param-test.h +1444 -0
  808. mypyc/external/googletest/include/gtest/gtest-param-test.h.pump +510 -0
  809. mypyc/external/googletest/include/gtest/gtest-printers.h +993 -0
  810. mypyc/external/googletest/include/gtest/gtest-spi.h +232 -0
  811. mypyc/external/googletest/include/gtest/gtest-test-part.h +179 -0
  812. mypyc/external/googletest/include/gtest/gtest-typed-test.h +263 -0
  813. mypyc/external/googletest/include/gtest/gtest.h +2236 -0
  814. mypyc/external/googletest/include/gtest/gtest_pred_impl.h +358 -0
  815. mypyc/external/googletest/include/gtest/gtest_prod.h +58 -0
  816. mypyc/external/googletest/include/gtest/internal/custom/gtest-port.h +69 -0
  817. mypyc/external/googletest/include/gtest/internal/custom/gtest-printers.h +42 -0
  818. mypyc/external/googletest/include/gtest/internal/custom/gtest.h +41 -0
  819. mypyc/external/googletest/include/gtest/internal/gtest-death-test-internal.h +319 -0
  820. mypyc/external/googletest/include/gtest/internal/gtest-filepath.h +206 -0
  821. mypyc/external/googletest/include/gtest/internal/gtest-internal.h +1238 -0
  822. mypyc/external/googletest/include/gtest/internal/gtest-linked_ptr.h +243 -0
  823. mypyc/external/googletest/include/gtest/internal/gtest-param-util-generated.h +5146 -0
  824. mypyc/external/googletest/include/gtest/internal/gtest-param-util-generated.h.pump +286 -0
  825. mypyc/external/googletest/include/gtest/internal/gtest-param-util.h +731 -0
  826. mypyc/external/googletest/include/gtest/internal/gtest-port-arch.h +93 -0
  827. mypyc/external/googletest/include/gtest/internal/gtest-port.h +2560 -0
  828. mypyc/external/googletest/include/gtest/internal/gtest-string.h +167 -0
  829. mypyc/external/googletest/include/gtest/internal/gtest-tuple.h +1020 -0
  830. mypyc/external/googletest/include/gtest/internal/gtest-tuple.h.pump +347 -0
  831. mypyc/external/googletest/include/gtest/internal/gtest-type-util.h +3331 -0
  832. mypyc/external/googletest/include/gtest/internal/gtest-type-util.h.pump +297 -0
  833. mypyc/external/googletest/make/Makefile +61 -0
  834. mypyc/external/googletest/src/gtest-all.cc +48 -0
  835. mypyc/external/googletest/src/gtest-death-test.cc +1342 -0
  836. mypyc/external/googletest/src/gtest-filepath.cc +387 -0
  837. mypyc/external/googletest/src/gtest-internal-inl.h +1183 -0
  838. mypyc/external/googletest/src/gtest-port.cc +1259 -0
  839. mypyc/external/googletest/src/gtest-printers.cc +373 -0
  840. mypyc/external/googletest/src/gtest-test-part.cc +110 -0
  841. mypyc/external/googletest/src/gtest-typed-test.cc +118 -0
  842. mypyc/external/googletest/src/gtest.cc +5388 -0
  843. mypyc/external/googletest/src/gtest_main.cc +38 -0
  844. mypyc/ir/__init__.py +0 -0
  845. mypyc/ir/class_ir.py +499 -0
  846. mypyc/ir/func_ir.py +370 -0
  847. mypyc/ir/module_ir.py +88 -0
  848. mypyc/ir/ops.py +1642 -0
  849. mypyc/ir/pprint.py +496 -0
  850. mypyc/ir/rtypes.py +1038 -0
  851. mypyc/irbuild/__init__.py +0 -0
  852. mypyc/irbuild/ast_helpers.py +120 -0
  853. mypyc/irbuild/builder.py +1398 -0
  854. mypyc/irbuild/callable_class.py +173 -0
  855. mypyc/irbuild/classdef.py +850 -0
  856. mypyc/irbuild/constant_fold.py +95 -0
  857. mypyc/irbuild/context.py +186 -0
  858. mypyc/irbuild/env_class.py +223 -0
  859. mypyc/irbuild/expression.py +1067 -0
  860. mypyc/irbuild/for_helpers.py +1046 -0
  861. mypyc/irbuild/format_str_tokenizer.py +250 -0
  862. mypyc/irbuild/function.py +1089 -0
  863. mypyc/irbuild/generator.py +346 -0
  864. mypyc/irbuild/ll_builder.py +2407 -0
  865. mypyc/irbuild/main.py +153 -0
  866. mypyc/irbuild/mapper.py +221 -0
  867. mypyc/irbuild/match.py +355 -0
  868. mypyc/irbuild/nonlocalcontrol.py +197 -0
  869. mypyc/irbuild/prebuildvisitor.py +203 -0
  870. mypyc/irbuild/prepare.py +609 -0
  871. mypyc/irbuild/specialize.py +822 -0
  872. mypyc/irbuild/statement.py +1017 -0
  873. mypyc/irbuild/targets.py +57 -0
  874. mypyc/irbuild/util.py +189 -0
  875. mypyc/irbuild/visitor.py +401 -0
  876. mypyc/irbuild/vtable.py +82 -0
  877. mypyc/lib-rt/CPy.h +643 -0
  878. mypyc/lib-rt/bytes_ops.c +143 -0
  879. mypyc/lib-rt/dict_ops.c +446 -0
  880. mypyc/lib-rt/exc_ops.c +259 -0
  881. mypyc/lib-rt/float_ops.c +192 -0
  882. mypyc/lib-rt/generic_ops.c +64 -0
  883. mypyc/lib-rt/getargs.c +450 -0
  884. mypyc/lib-rt/getargsfast.c +569 -0
  885. mypyc/lib-rt/init.c +13 -0
  886. mypyc/lib-rt/int_ops.c +803 -0
  887. mypyc/lib-rt/list_ops.c +335 -0
  888. mypyc/lib-rt/misc_ops.c +942 -0
  889. mypyc/lib-rt/module_shim.tmpl +18 -0
  890. mypyc/lib-rt/mypyc_util.h +118 -0
  891. mypyc/lib-rt/pythoncapi_compat.h +497 -0
  892. mypyc/lib-rt/pythonsupport.h +548 -0
  893. mypyc/lib-rt/set_ops.c +17 -0
  894. mypyc/lib-rt/setup.py +70 -0
  895. mypyc/lib-rt/str_ops.c +241 -0
  896. mypyc/lib-rt/test_capi.cc +585 -0
  897. mypyc/lib-rt/tuple_ops.c +61 -0
  898. mypyc/namegen.py +115 -0
  899. mypyc/options.py +32 -0
  900. mypyc/primitives/__init__.py +0 -0
  901. mypyc/primitives/bytes_ops.py +101 -0
  902. mypyc/primitives/dict_ops.py +325 -0
  903. mypyc/primitives/exc_ops.py +101 -0
  904. mypyc/primitives/float_ops.py +168 -0
  905. mypyc/primitives/generic_ops.py +384 -0
  906. mypyc/primitives/int_ops.py +304 -0
  907. mypyc/primitives/list_ops.py +300 -0
  908. mypyc/primitives/misc_ops.py +241 -0
  909. mypyc/primitives/registry.py +323 -0
  910. mypyc/primitives/set_ops.py +121 -0
  911. mypyc/primitives/str_ops.py +229 -0
  912. mypyc/primitives/tuple_ops.py +83 -0
  913. mypyc/rt_subtype.py +77 -0
  914. mypyc/sametype.py +83 -0
  915. mypyc/subtype.py +88 -0
  916. mypyc/test/__init__.py +0 -0
  917. mypyc/test/config.py +13 -0
  918. mypyc/test/test_alwaysdefined.py +46 -0
  919. mypyc/test/test_analysis.py +77 -0
  920. mypyc/test/test_cheader.py +45 -0
  921. mypyc/test/test_commandline.py +82 -0
  922. mypyc/test/test_emit.py +69 -0
  923. mypyc/test/test_emitclass.py +35 -0
  924. mypyc/test/test_emitfunc.py +926 -0
  925. mypyc/test/test_emitwrapper.py +60 -0
  926. mypyc/test/test_exceptions.py +56 -0
  927. mypyc/test/test_external.py +49 -0
  928. mypyc/test/test_irbuild.py +87 -0
  929. mypyc/test/test_ircheck.py +199 -0
  930. mypyc/test/test_literals.py +90 -0
  931. mypyc/test/test_namegen.py +48 -0
  932. mypyc/test/test_pprint.py +42 -0
  933. mypyc/test/test_rarray.py +48 -0
  934. mypyc/test/test_refcount.py +59 -0
  935. mypyc/test/test_run.py +426 -0
  936. mypyc/test/test_serialization.py +108 -0
  937. mypyc/test/test_struct.py +112 -0
  938. mypyc/test/test_tuplename.py +33 -0
  939. mypyc/test/test_typeops.py +97 -0
  940. mypyc/test/testutil.py +283 -0
  941. mypyc/test-data/alwaysdefined.test +732 -0
  942. mypyc/test-data/analysis.test +603 -0
  943. mypyc/test-data/commandline.test +245 -0
  944. mypyc/test-data/driver/driver.py +48 -0
  945. mypyc/test-data/exceptions-freq.test +125 -0
  946. mypyc/test-data/exceptions.test +713 -0
  947. mypyc/test-data/fixtures/ir.py +373 -0
  948. mypyc/test-data/fixtures/testutil.py +103 -0
  949. mypyc/test-data/fixtures/typing-full.pyi +169 -0
  950. mypyc/test-data/irbuild-any.test +238 -0
  951. mypyc/test-data/irbuild-basic.test +3690 -0
  952. mypyc/test-data/irbuild-bool.test +463 -0
  953. mypyc/test-data/irbuild-bytes.test +184 -0
  954. mypyc/test-data/irbuild-classes.test +1303 -0
  955. mypyc/test-data/irbuild-constant-fold.test +480 -0
  956. mypyc/test-data/irbuild-dict.test +585 -0
  957. mypyc/test-data/irbuild-dunders.test +223 -0
  958. mypyc/test-data/irbuild-float.test +497 -0
  959. mypyc/test-data/irbuild-generics.test +151 -0
  960. mypyc/test-data/irbuild-glue-methods.test +437 -0
  961. mypyc/test-data/irbuild-i16.test +526 -0
  962. mypyc/test-data/irbuild-i32.test +598 -0
  963. mypyc/test-data/irbuild-i64.test +2164 -0
  964. mypyc/test-data/irbuild-int.test +235 -0
  965. mypyc/test-data/irbuild-isinstance.test +109 -0
  966. mypyc/test-data/irbuild-lists.test +547 -0
  967. mypyc/test-data/irbuild-match.test +1708 -0
  968. mypyc/test-data/irbuild-math.test +64 -0
  969. mypyc/test-data/irbuild-nested.test +807 -0
  970. mypyc/test-data/irbuild-optional.test +536 -0
  971. mypyc/test-data/irbuild-set.test +838 -0
  972. mypyc/test-data/irbuild-singledispatch.test +257 -0
  973. mypyc/test-data/irbuild-statements.test +1149 -0
  974. mypyc/test-data/irbuild-str.test +314 -0
  975. mypyc/test-data/irbuild-strip-asserts.test +12 -0
  976. mypyc/test-data/irbuild-try.test +523 -0
  977. mypyc/test-data/irbuild-tuple.test +462 -0
  978. mypyc/test-data/irbuild-u8.test +543 -0
  979. mypyc/test-data/irbuild-unreachable.test +241 -0
  980. mypyc/test-data/irbuild-vectorcall.test +153 -0
  981. mypyc/test-data/refcount.test +1588 -0
  982. mypyc/test-data/run-async.test +145 -0
  983. mypyc/test-data/run-attrs.test +318 -0
  984. mypyc/test-data/run-bench.test +196 -0
  985. mypyc/test-data/run-bools.test +229 -0
  986. mypyc/test-data/run-bytes.test +302 -0
  987. mypyc/test-data/run-classes.test +2505 -0
  988. mypyc/test-data/run-dicts.test +334 -0
  989. mypyc/test-data/run-dunders.test +945 -0
  990. mypyc/test-data/run-exceptions.test +448 -0
  991. mypyc/test-data/run-floats.test +516 -0
  992. mypyc/test-data/run-functions.test +1310 -0
  993. mypyc/test-data/run-generators.test +681 -0
  994. mypyc/test-data/run-i16.test +338 -0
  995. mypyc/test-data/run-i32.test +336 -0
  996. mypyc/test-data/run-i64.test +1519 -0
  997. mypyc/test-data/run-imports.test +265 -0
  998. mypyc/test-data/run-integers.test +540 -0
  999. mypyc/test-data/run-lists.test +411 -0
  1000. mypyc/test-data/run-loops.test +485 -0
  1001. mypyc/test-data/run-match.test +283 -0
  1002. mypyc/test-data/run-math.test +106 -0
  1003. mypyc/test-data/run-misc.test +1168 -0
  1004. mypyc/test-data/run-multimodule.test +887 -0
  1005. mypyc/test-data/run-mypy-sim.test +138 -0
  1006. mypyc/test-data/run-primitives.test +375 -0
  1007. mypyc/test-data/run-python37.test +159 -0
  1008. mypyc/test-data/run-python38.test +88 -0
  1009. mypyc/test-data/run-sets.test +150 -0
  1010. mypyc/test-data/run-singledispatch.test +698 -0
  1011. mypyc/test-data/run-strings.test +641 -0
  1012. mypyc/test-data/run-traits.test +411 -0
  1013. mypyc/test-data/run-tuples.test +258 -0
  1014. mypyc/test-data/run-u8.test +303 -0
  1015. mypyc/transform/__init__.py +0 -0
  1016. mypyc/transform/exceptions.py +182 -0
  1017. mypyc/transform/refcount.py +294 -0
  1018. mypyc/transform/uninit.py +190 -0
@@ -0,0 +1,846 @@
1
+ """Code generation for native function bodies."""
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Final
6
+
7
+ from mypyc.analysis.blockfreq import frequently_executed_blocks
8
+ from mypyc.codegen.emit import DEBUG_ERRORS, Emitter, TracebackAndGotoHandler, c_array_initializer
9
+ from mypyc.common import MODULE_PREFIX, NATIVE_PREFIX, REG_PREFIX, STATIC_PREFIX, TYPE_PREFIX
10
+ from mypyc.ir.class_ir import ClassIR
11
+ from mypyc.ir.func_ir import FUNC_CLASSMETHOD, FUNC_STATICMETHOD, FuncDecl, FuncIR, all_values
12
+ from mypyc.ir.ops import (
13
+ ERR_FALSE,
14
+ NAMESPACE_MODULE,
15
+ NAMESPACE_STATIC,
16
+ NAMESPACE_TYPE,
17
+ Assign,
18
+ AssignMulti,
19
+ BasicBlock,
20
+ Box,
21
+ Branch,
22
+ Call,
23
+ CallC,
24
+ Cast,
25
+ ComparisonOp,
26
+ ControlOp,
27
+ DecRef,
28
+ Extend,
29
+ Float,
30
+ FloatComparisonOp,
31
+ FloatNeg,
32
+ FloatOp,
33
+ GetAttr,
34
+ GetElementPtr,
35
+ Goto,
36
+ IncRef,
37
+ InitStatic,
38
+ Integer,
39
+ IntOp,
40
+ KeepAlive,
41
+ LoadAddress,
42
+ LoadErrorValue,
43
+ LoadGlobal,
44
+ LoadLiteral,
45
+ LoadMem,
46
+ LoadStatic,
47
+ MethodCall,
48
+ Op,
49
+ OpVisitor,
50
+ RaiseStandardError,
51
+ Register,
52
+ Return,
53
+ SetAttr,
54
+ SetMem,
55
+ Truncate,
56
+ TupleGet,
57
+ TupleSet,
58
+ Unborrow,
59
+ Unbox,
60
+ Unreachable,
61
+ Value,
62
+ )
63
+ from mypyc.ir.pprint import generate_names_for_ir
64
+ from mypyc.ir.rtypes import (
65
+ RArray,
66
+ RStruct,
67
+ RTuple,
68
+ RType,
69
+ is_int32_rprimitive,
70
+ is_int64_rprimitive,
71
+ is_int_rprimitive,
72
+ is_pointer_rprimitive,
73
+ is_tagged,
74
+ )
75
+
76
+
77
+ def native_function_type(fn: FuncIR, emitter: Emitter) -> str:
78
+ args = ", ".join(emitter.ctype(arg.type) for arg in fn.args) or "void"
79
+ ret = emitter.ctype(fn.ret_type)
80
+ return f"{ret} (*)({args})"
81
+
82
+
83
+ def native_function_header(fn: FuncDecl, emitter: Emitter) -> str:
84
+ args = []
85
+ for arg in fn.sig.args:
86
+ args.append(f"{emitter.ctype_spaced(arg.type)}{REG_PREFIX}{arg.name}")
87
+
88
+ return "{ret_type}{name}({args})".format(
89
+ ret_type=emitter.ctype_spaced(fn.sig.ret_type),
90
+ name=emitter.native_function_name(fn),
91
+ args=", ".join(args) or "void",
92
+ )
93
+
94
+
95
+ def generate_native_function(
96
+ fn: FuncIR, emitter: Emitter, source_path: str, module_name: str
97
+ ) -> None:
98
+ declarations = Emitter(emitter.context)
99
+ names = generate_names_for_ir(fn.arg_regs, fn.blocks)
100
+ body = Emitter(emitter.context, names)
101
+ visitor = FunctionEmitterVisitor(body, declarations, source_path, module_name)
102
+
103
+ declarations.emit_line(f"{native_function_header(fn.decl, emitter)} {{")
104
+ body.indent()
105
+
106
+ for r in all_values(fn.arg_regs, fn.blocks):
107
+ if isinstance(r.type, RTuple):
108
+ emitter.declare_tuple_struct(r.type)
109
+ if isinstance(r.type, RArray):
110
+ continue # Special: declared on first assignment
111
+
112
+ if r in fn.arg_regs:
113
+ continue # Skip the arguments
114
+
115
+ ctype = emitter.ctype_spaced(r.type)
116
+ init = ""
117
+ declarations.emit_line(
118
+ "{ctype}{prefix}{name}{init};".format(
119
+ ctype=ctype, prefix=REG_PREFIX, name=names[r], init=init
120
+ )
121
+ )
122
+
123
+ # Before we emit the blocks, give them all labels
124
+ blocks = fn.blocks
125
+ for i, block in enumerate(blocks):
126
+ block.label = i
127
+
128
+ # Find blocks that are never jumped to or are only jumped to from the
129
+ # block directly above it. This allows for more labels and gotos to be
130
+ # eliminated during code generation.
131
+ for block in fn.blocks:
132
+ terminator = block.terminator
133
+ assert isinstance(terminator, ControlOp)
134
+
135
+ for target in terminator.targets():
136
+ is_next_block = target.label == block.label + 1
137
+
138
+ # Always emit labels for GetAttr error checks since the emit code that
139
+ # generates them will add instructions between the branch and the
140
+ # next label, causing the label to be wrongly removed. A better
141
+ # solution would be to change the IR so that it adds a basic block
142
+ # inbetween the calls.
143
+ is_problematic_op = isinstance(terminator, Branch) and any(
144
+ isinstance(s, GetAttr) for s in terminator.sources()
145
+ )
146
+
147
+ if not is_next_block or is_problematic_op:
148
+ fn.blocks[target.label].referenced = True
149
+
150
+ common = frequently_executed_blocks(fn.blocks[0])
151
+
152
+ for i in range(len(blocks)):
153
+ block = blocks[i]
154
+ visitor.rare = block not in common
155
+ next_block = None
156
+ if i + 1 < len(blocks):
157
+ next_block = blocks[i + 1]
158
+ body.emit_label(block)
159
+ visitor.next_block = next_block
160
+
161
+ ops = block.ops
162
+ visitor.ops = ops
163
+ visitor.op_index = 0
164
+ while visitor.op_index < len(ops):
165
+ ops[visitor.op_index].accept(visitor)
166
+ visitor.op_index += 1
167
+
168
+ body.emit_line("}")
169
+
170
+ emitter.emit_from_emitter(declarations)
171
+ emitter.emit_from_emitter(body)
172
+
173
+
174
+ class FunctionEmitterVisitor(OpVisitor[None]):
175
+ def __init__(
176
+ self, emitter: Emitter, declarations: Emitter, source_path: str, module_name: str
177
+ ) -> None:
178
+ self.emitter = emitter
179
+ self.names = emitter.names
180
+ self.declarations = declarations
181
+ self.source_path = source_path
182
+ self.module_name = module_name
183
+ self.literals = emitter.context.literals
184
+ self.rare = False
185
+ # Next basic block to be processed after the current one (if any), set by caller
186
+ self.next_block: BasicBlock | None = None
187
+ # Ops in the basic block currently being processed, set by caller
188
+ self.ops: list[Op] = []
189
+ # Current index within ops; visit methods can increment this to skip/merge ops
190
+ self.op_index = 0
191
+
192
+ def temp_name(self) -> str:
193
+ return self.emitter.temp_name()
194
+
195
+ def visit_goto(self, op: Goto) -> None:
196
+ if op.label is not self.next_block:
197
+ self.emit_line("goto %s;" % self.label(op.label))
198
+
199
+ def visit_branch(self, op: Branch) -> None:
200
+ true, false = op.true, op.false
201
+ negated = op.negated
202
+ negated_rare = False
203
+ if true is self.next_block and op.traceback_entry is None:
204
+ # Switch true/false since it avoids an else block.
205
+ true, false = false, true
206
+ negated = not negated
207
+ negated_rare = True
208
+
209
+ neg = "!" if negated else ""
210
+ cond = ""
211
+ if op.op == Branch.BOOL:
212
+ expr_result = self.reg(op.value)
213
+ cond = f"{neg}{expr_result}"
214
+ elif op.op == Branch.IS_ERROR:
215
+ typ = op.value.type
216
+ compare = "!=" if negated else "=="
217
+ if isinstance(typ, RTuple):
218
+ # TODO: What about empty tuple?
219
+ cond = self.emitter.tuple_undefined_check_cond(
220
+ typ, self.reg(op.value), self.c_error_value, compare
221
+ )
222
+ else:
223
+ cond = f"{self.reg(op.value)} {compare} {self.c_error_value(typ)}"
224
+ else:
225
+ assert False, "Invalid branch"
226
+
227
+ # For error checks, tell the compiler the branch is unlikely
228
+ if op.traceback_entry is not None or op.rare:
229
+ if not negated_rare:
230
+ cond = f"unlikely({cond})"
231
+ else:
232
+ cond = f"likely({cond})"
233
+
234
+ if false is self.next_block:
235
+ if op.traceback_entry is None:
236
+ if true is not self.next_block:
237
+ self.emit_line(f"if ({cond}) goto {self.label(true)};")
238
+ else:
239
+ self.emit_line(f"if ({cond}) {{")
240
+ self.emit_traceback(op)
241
+ self.emit_lines("goto %s;" % self.label(true), "}")
242
+ else:
243
+ self.emit_line(f"if ({cond}) {{")
244
+ self.emit_traceback(op)
245
+
246
+ if true is not self.next_block:
247
+ self.emit_line("goto %s;" % self.label(true))
248
+
249
+ self.emit_lines("} else", " goto %s;" % self.label(false))
250
+
251
+ def visit_return(self, op: Return) -> None:
252
+ value_str = self.reg(op.value)
253
+ self.emit_line("return %s;" % value_str)
254
+
255
+ def visit_tuple_set(self, op: TupleSet) -> None:
256
+ dest = self.reg(op)
257
+ tuple_type = op.tuple_type
258
+ self.emitter.declare_tuple_struct(tuple_type)
259
+ if len(op.items) == 0: # empty tuple
260
+ self.emit_line(f"{dest}.empty_struct_error_flag = 0;")
261
+ else:
262
+ for i, item in enumerate(op.items):
263
+ self.emit_line(f"{dest}.f{i} = {self.reg(item)};")
264
+
265
+ def visit_assign(self, op: Assign) -> None:
266
+ dest = self.reg(op.dest)
267
+ src = self.reg(op.src)
268
+ # clang whines about self assignment (which we might generate
269
+ # for some casts), so don't emit it.
270
+ if dest != src:
271
+ # We sometimes assign from an integer prepresentation of a pointer
272
+ # to a real pointer, and C compilers insist on a cast.
273
+ if op.src.type.is_unboxed and not op.dest.type.is_unboxed:
274
+ src = f"(void *){src}"
275
+ self.emit_line(f"{dest} = {src};")
276
+
277
+ def visit_assign_multi(self, op: AssignMulti) -> None:
278
+ typ = op.dest.type
279
+ assert isinstance(typ, RArray)
280
+ dest = self.reg(op.dest)
281
+ # RArray values can only be assigned to once, so we can always
282
+ # declare them on initialization.
283
+ self.emit_line(
284
+ "%s%s[%d] = %s;"
285
+ % (
286
+ self.emitter.ctype_spaced(typ.item_type),
287
+ dest,
288
+ len(op.src),
289
+ c_array_initializer([self.reg(s) for s in op.src], indented=True),
290
+ )
291
+ )
292
+
293
+ def visit_load_error_value(self, op: LoadErrorValue) -> None:
294
+ if isinstance(op.type, RTuple):
295
+ values = [self.c_undefined_value(item) for item in op.type.types]
296
+ tmp = self.temp_name()
297
+ self.emit_line("{} {} = {{ {} }};".format(self.ctype(op.type), tmp, ", ".join(values)))
298
+ self.emit_line(f"{self.reg(op)} = {tmp};")
299
+ else:
300
+ self.emit_line(f"{self.reg(op)} = {self.c_error_value(op.type)};")
301
+
302
+ def visit_load_literal(self, op: LoadLiteral) -> None:
303
+ index = self.literals.literal_index(op.value)
304
+ if not is_int_rprimitive(op.type):
305
+ self.emit_line("%s = CPyStatics[%d];" % (self.reg(op), index), ann=op.value)
306
+ else:
307
+ self.emit_line(
308
+ "%s = (CPyTagged)CPyStatics[%d] | 1;" % (self.reg(op), index), ann=op.value
309
+ )
310
+
311
+ def get_attr_expr(self, obj: str, op: GetAttr | SetAttr, decl_cl: ClassIR) -> str:
312
+ """Generate attribute accessor for normal (non-property) access.
313
+
314
+ This either has a form like obj->attr_name for attributes defined in non-trait
315
+ classes, and *(obj + attr_offset) for attributes defined by traits. We also
316
+ insert all necessary C casts here.
317
+ """
318
+ cast = f"({op.class_type.struct_name(self.emitter.names)} *)"
319
+ if decl_cl.is_trait and op.class_type.class_ir.is_trait:
320
+ # For pure trait access find the offset first, offsets
321
+ # are ordered by attribute position in the cl.attributes dict.
322
+ # TODO: pre-calculate the mapping to make this faster.
323
+ trait_attr_index = list(decl_cl.attributes).index(op.attr)
324
+ # TODO: reuse these names somehow?
325
+ offset = self.emitter.temp_name()
326
+ self.declarations.emit_line(f"size_t {offset};")
327
+ self.emitter.emit_line(
328
+ "{} = {};".format(
329
+ offset,
330
+ "CPy_FindAttrOffset({}, {}, {})".format(
331
+ self.emitter.type_struct_name(decl_cl),
332
+ f"({cast}{obj})->vtable",
333
+ trait_attr_index,
334
+ ),
335
+ )
336
+ )
337
+ attr_cast = f"({self.ctype(op.class_type.attr_type(op.attr))} *)"
338
+ return f"*{attr_cast}((char *){obj} + {offset})"
339
+ else:
340
+ # Cast to something non-trait. Note: for this to work, all struct
341
+ # members for non-trait classes must obey monotonic linear growth.
342
+ if op.class_type.class_ir.is_trait:
343
+ assert not decl_cl.is_trait
344
+ cast = f"({decl_cl.struct_name(self.emitter.names)} *)"
345
+ return f"({cast}{obj})->{self.emitter.attr(op.attr)}"
346
+
347
+ def visit_get_attr(self, op: GetAttr) -> None:
348
+ dest = self.reg(op)
349
+ obj = self.reg(op.obj)
350
+ rtype = op.class_type
351
+ cl = rtype.class_ir
352
+ attr_rtype, decl_cl = cl.attr_details(op.attr)
353
+ prefer_method = cl.is_trait and attr_rtype.error_overlap
354
+ if cl.get_method(op.attr, prefer_method=prefer_method):
355
+ # Properties are essentially methods, so use vtable access for them.
356
+ version = "_TRAIT" if cl.is_trait else ""
357
+ self.emit_line(
358
+ "%s = CPY_GET_ATTR%s(%s, %s, %d, %s, %s); /* %s */"
359
+ % (
360
+ dest,
361
+ version,
362
+ obj,
363
+ self.emitter.type_struct_name(rtype.class_ir),
364
+ rtype.getter_index(op.attr),
365
+ rtype.struct_name(self.names),
366
+ self.ctype(rtype.attr_type(op.attr)),
367
+ op.attr,
368
+ )
369
+ )
370
+ else:
371
+ # Otherwise, use direct or offset struct access.
372
+ attr_expr = self.get_attr_expr(obj, op, decl_cl)
373
+ self.emitter.emit_line(f"{dest} = {attr_expr};")
374
+ always_defined = cl.is_always_defined(op.attr)
375
+ merged_branch = None
376
+ if not always_defined:
377
+ self.emitter.emit_undefined_attr_check(
378
+ attr_rtype, dest, "==", obj, op.attr, cl, unlikely=True
379
+ )
380
+ branch = self.next_branch()
381
+ if branch is not None:
382
+ if (
383
+ branch.value is op
384
+ and branch.op == Branch.IS_ERROR
385
+ and branch.traceback_entry is not None
386
+ and not branch.negated
387
+ ):
388
+ # Generate code for the following branch here to avoid
389
+ # redundant branches in the generated code.
390
+ self.emit_attribute_error(branch, cl.name, op.attr)
391
+ self.emit_line("goto %s;" % self.label(branch.true))
392
+ merged_branch = branch
393
+ self.emitter.emit_line("}")
394
+ if not merged_branch:
395
+ exc_class = "PyExc_AttributeError"
396
+ self.emitter.emit_line(
397
+ 'PyErr_SetString({}, "attribute {} of {} undefined");'.format(
398
+ exc_class, repr(op.attr), repr(cl.name)
399
+ )
400
+ )
401
+
402
+ if attr_rtype.is_refcounted and not op.is_borrowed:
403
+ if not merged_branch and not always_defined:
404
+ self.emitter.emit_line("} else {")
405
+ self.emitter.emit_inc_ref(dest, attr_rtype)
406
+ if merged_branch:
407
+ if merged_branch.false is not self.next_block:
408
+ self.emit_line("goto %s;" % self.label(merged_branch.false))
409
+ self.op_index += 1
410
+ elif not always_defined:
411
+ self.emitter.emit_line("}")
412
+
413
+ def next_branch(self) -> Branch | None:
414
+ if self.op_index + 1 < len(self.ops):
415
+ next_op = self.ops[self.op_index + 1]
416
+ if isinstance(next_op, Branch):
417
+ return next_op
418
+ return None
419
+
420
+ def visit_set_attr(self, op: SetAttr) -> None:
421
+ if op.error_kind == ERR_FALSE:
422
+ dest = self.reg(op)
423
+ obj = self.reg(op.obj)
424
+ src = self.reg(op.src)
425
+ rtype = op.class_type
426
+ cl = rtype.class_ir
427
+ attr_rtype, decl_cl = cl.attr_details(op.attr)
428
+ if cl.get_method(op.attr):
429
+ # Again, use vtable access for properties...
430
+ assert not op.is_init and op.error_kind == ERR_FALSE, "%s %d %d %s" % (
431
+ op.attr,
432
+ op.is_init,
433
+ op.error_kind,
434
+ rtype,
435
+ )
436
+ version = "_TRAIT" if cl.is_trait else ""
437
+ self.emit_line(
438
+ "%s = CPY_SET_ATTR%s(%s, %s, %d, %s, %s, %s); /* %s */"
439
+ % (
440
+ dest,
441
+ version,
442
+ obj,
443
+ self.emitter.type_struct_name(rtype.class_ir),
444
+ rtype.setter_index(op.attr),
445
+ src,
446
+ rtype.struct_name(self.names),
447
+ self.ctype(rtype.attr_type(op.attr)),
448
+ op.attr,
449
+ )
450
+ )
451
+ else:
452
+ # ...and struct access for normal attributes.
453
+ attr_expr = self.get_attr_expr(obj, op, decl_cl)
454
+ if not op.is_init and attr_rtype.is_refcounted:
455
+ # This is not an initialization (where we know that the attribute was
456
+ # previously undefined), so decref the old value.
457
+ always_defined = cl.is_always_defined(op.attr)
458
+ if not always_defined:
459
+ self.emitter.emit_undefined_attr_check(
460
+ attr_rtype, attr_expr, "!=", obj, op.attr, cl
461
+ )
462
+ self.emitter.emit_dec_ref(attr_expr, attr_rtype)
463
+ if not always_defined:
464
+ self.emitter.emit_line("}")
465
+ elif attr_rtype.error_overlap and not cl.is_always_defined(op.attr):
466
+ # If there is overlap with the error value, update bitmap to mark
467
+ # attribute as defined.
468
+ self.emitter.emit_attr_bitmap_set(src, obj, attr_rtype, cl, op.attr)
469
+
470
+ # This steals the reference to src, so we don't need to increment the arg
471
+ self.emitter.emit_line(f"{attr_expr} = {src};")
472
+ if op.error_kind == ERR_FALSE:
473
+ self.emitter.emit_line(f"{dest} = 1;")
474
+
475
+ PREFIX_MAP: Final = {
476
+ NAMESPACE_STATIC: STATIC_PREFIX,
477
+ NAMESPACE_TYPE: TYPE_PREFIX,
478
+ NAMESPACE_MODULE: MODULE_PREFIX,
479
+ }
480
+
481
+ def visit_load_static(self, op: LoadStatic) -> None:
482
+ dest = self.reg(op)
483
+ prefix = self.PREFIX_MAP[op.namespace]
484
+ name = self.emitter.static_name(op.identifier, op.module_name, prefix)
485
+ if op.namespace == NAMESPACE_TYPE:
486
+ name = "(PyObject *)%s" % name
487
+ self.emit_line(f"{dest} = {name};", ann=op.ann)
488
+
489
+ def visit_init_static(self, op: InitStatic) -> None:
490
+ value = self.reg(op.value)
491
+ prefix = self.PREFIX_MAP[op.namespace]
492
+ name = self.emitter.static_name(op.identifier, op.module_name, prefix)
493
+ if op.namespace == NAMESPACE_TYPE:
494
+ value = "(PyTypeObject *)%s" % value
495
+ self.emit_line(f"{name} = {value};")
496
+ self.emit_inc_ref(name, op.value.type)
497
+
498
+ def visit_tuple_get(self, op: TupleGet) -> None:
499
+ dest = self.reg(op)
500
+ src = self.reg(op.src)
501
+ self.emit_line(f"{dest} = {src}.f{op.index};")
502
+ if not op.is_borrowed:
503
+ self.emit_inc_ref(dest, op.type)
504
+
505
+ def get_dest_assign(self, dest: Value) -> str:
506
+ if not dest.is_void:
507
+ return self.reg(dest) + " = "
508
+ else:
509
+ return ""
510
+
511
+ def visit_call(self, op: Call) -> None:
512
+ """Call native function."""
513
+ dest = self.get_dest_assign(op)
514
+ args = ", ".join(self.reg(arg) for arg in op.args)
515
+ lib = self.emitter.get_group_prefix(op.fn)
516
+ cname = op.fn.cname(self.names)
517
+ self.emit_line(f"{dest}{lib}{NATIVE_PREFIX}{cname}({args});")
518
+
519
+ def visit_method_call(self, op: MethodCall) -> None:
520
+ """Call native method."""
521
+ dest = self.get_dest_assign(op)
522
+ obj = self.reg(op.obj)
523
+
524
+ rtype = op.receiver_type
525
+ class_ir = rtype.class_ir
526
+ name = op.method
527
+ method = rtype.class_ir.get_method(name)
528
+ assert method is not None
529
+
530
+ # Can we call the method directly, bypassing vtable?
531
+ is_direct = class_ir.is_method_final(name)
532
+
533
+ # The first argument gets omitted for static methods and
534
+ # turned into the class for class methods
535
+ obj_args = (
536
+ []
537
+ if method.decl.kind == FUNC_STATICMETHOD
538
+ else [f"(PyObject *)Py_TYPE({obj})"] if method.decl.kind == FUNC_CLASSMETHOD else [obj]
539
+ )
540
+ args = ", ".join(obj_args + [self.reg(arg) for arg in op.args])
541
+ mtype = native_function_type(method, self.emitter)
542
+ version = "_TRAIT" if rtype.class_ir.is_trait else ""
543
+ if is_direct:
544
+ # Directly call method, without going through the vtable.
545
+ lib = self.emitter.get_group_prefix(method.decl)
546
+ self.emit_line(f"{dest}{lib}{NATIVE_PREFIX}{method.cname(self.names)}({args});")
547
+ else:
548
+ # Call using vtable.
549
+ method_idx = rtype.method_index(name)
550
+ self.emit_line(
551
+ "{}CPY_GET_METHOD{}({}, {}, {}, {}, {})({}); /* {} */".format(
552
+ dest,
553
+ version,
554
+ obj,
555
+ self.emitter.type_struct_name(rtype.class_ir),
556
+ method_idx,
557
+ rtype.struct_name(self.names),
558
+ mtype,
559
+ args,
560
+ op.method,
561
+ )
562
+ )
563
+
564
+ def visit_inc_ref(self, op: IncRef) -> None:
565
+ src = self.reg(op.src)
566
+ self.emit_inc_ref(src, op.src.type)
567
+
568
+ def visit_dec_ref(self, op: DecRef) -> None:
569
+ src = self.reg(op.src)
570
+ self.emit_dec_ref(src, op.src.type, is_xdec=op.is_xdec)
571
+
572
+ def visit_box(self, op: Box) -> None:
573
+ self.emitter.emit_box(self.reg(op.src), self.reg(op), op.src.type, can_borrow=True)
574
+
575
+ def visit_cast(self, op: Cast) -> None:
576
+ branch = self.next_branch()
577
+ handler = None
578
+ if branch is not None:
579
+ if (
580
+ branch.value is op
581
+ and branch.op == Branch.IS_ERROR
582
+ and branch.traceback_entry is not None
583
+ and not branch.negated
584
+ and branch.false is self.next_block
585
+ ):
586
+ # Generate code also for the following branch here to avoid
587
+ # redundant branches in the generated code.
588
+ handler = TracebackAndGotoHandler(
589
+ self.label(branch.true),
590
+ self.source_path,
591
+ self.module_name,
592
+ branch.traceback_entry,
593
+ )
594
+ self.op_index += 1
595
+
596
+ self.emitter.emit_cast(
597
+ self.reg(op.src), self.reg(op), op.type, src_type=op.src.type, error=handler
598
+ )
599
+
600
+ def visit_unbox(self, op: Unbox) -> None:
601
+ self.emitter.emit_unbox(self.reg(op.src), self.reg(op), op.type)
602
+
603
+ def visit_unreachable(self, op: Unreachable) -> None:
604
+ self.emitter.emit_line("CPy_Unreachable();")
605
+
606
+ def visit_raise_standard_error(self, op: RaiseStandardError) -> None:
607
+ # TODO: Better escaping of backspaces and such
608
+ if op.value is not None:
609
+ if isinstance(op.value, str):
610
+ message = op.value.replace('"', '\\"')
611
+ self.emitter.emit_line(f'PyErr_SetString(PyExc_{op.class_name}, "{message}");')
612
+ elif isinstance(op.value, Value):
613
+ self.emitter.emit_line(
614
+ "PyErr_SetObject(PyExc_{}, {});".format(
615
+ op.class_name, self.emitter.reg(op.value)
616
+ )
617
+ )
618
+ else:
619
+ assert False, "op value type must be either str or Value"
620
+ else:
621
+ self.emitter.emit_line(f"PyErr_SetNone(PyExc_{op.class_name});")
622
+ self.emitter.emit_line(f"{self.reg(op)} = 0;")
623
+
624
+ def visit_call_c(self, op: CallC) -> None:
625
+ if op.is_void:
626
+ dest = ""
627
+ else:
628
+ dest = self.get_dest_assign(op)
629
+ args = ", ".join(self.reg(arg) for arg in op.args)
630
+ self.emitter.emit_line(f"{dest}{op.function_name}({args});")
631
+
632
+ def visit_truncate(self, op: Truncate) -> None:
633
+ dest = self.reg(op)
634
+ value = self.reg(op.src)
635
+ # for C backend the generated code are straight assignments
636
+ self.emit_line(f"{dest} = {value};")
637
+
638
+ def visit_extend(self, op: Extend) -> None:
639
+ dest = self.reg(op)
640
+ value = self.reg(op.src)
641
+ if op.signed:
642
+ src_cast = self.emit_signed_int_cast(op.src.type)
643
+ else:
644
+ src_cast = self.emit_unsigned_int_cast(op.src.type)
645
+ self.emit_line(f"{dest} = {src_cast}{value};")
646
+
647
+ def visit_load_global(self, op: LoadGlobal) -> None:
648
+ dest = self.reg(op)
649
+ self.emit_line(f"{dest} = {op.identifier};", ann=op.ann)
650
+
651
+ def visit_int_op(self, op: IntOp) -> None:
652
+ dest = self.reg(op)
653
+ lhs = self.reg(op.lhs)
654
+ rhs = self.reg(op.rhs)
655
+ if op.op == IntOp.RIGHT_SHIFT:
656
+ # Signed right shift
657
+ lhs = self.emit_signed_int_cast(op.lhs.type) + lhs
658
+ rhs = self.emit_signed_int_cast(op.rhs.type) + rhs
659
+ self.emit_line(f"{dest} = {lhs} {op.op_str[op.op]} {rhs};")
660
+
661
+ def visit_comparison_op(self, op: ComparisonOp) -> None:
662
+ dest = self.reg(op)
663
+ lhs = self.reg(op.lhs)
664
+ rhs = self.reg(op.rhs)
665
+ lhs_cast = ""
666
+ rhs_cast = ""
667
+ if op.op in (ComparisonOp.SLT, ComparisonOp.SGT, ComparisonOp.SLE, ComparisonOp.SGE):
668
+ # Always signed comparison op
669
+ lhs_cast = self.emit_signed_int_cast(op.lhs.type)
670
+ rhs_cast = self.emit_signed_int_cast(op.rhs.type)
671
+ elif op.op in (ComparisonOp.ULT, ComparisonOp.UGT, ComparisonOp.ULE, ComparisonOp.UGE):
672
+ # Always unsigned comparison op
673
+ lhs_cast = self.emit_unsigned_int_cast(op.lhs.type)
674
+ rhs_cast = self.emit_unsigned_int_cast(op.rhs.type)
675
+ elif isinstance(op.lhs, Integer) and op.lhs.value < 0:
676
+ # Force signed ==/!= with negative operand
677
+ rhs_cast = self.emit_signed_int_cast(op.rhs.type)
678
+ elif isinstance(op.rhs, Integer) and op.rhs.value < 0:
679
+ # Force signed ==/!= with negative operand
680
+ lhs_cast = self.emit_signed_int_cast(op.lhs.type)
681
+ self.emit_line(f"{dest} = {lhs_cast}{lhs} {op.op_str[op.op]} {rhs_cast}{rhs};")
682
+
683
+ def visit_float_op(self, op: FloatOp) -> None:
684
+ dest = self.reg(op)
685
+ lhs = self.reg(op.lhs)
686
+ rhs = self.reg(op.rhs)
687
+ if op.op != FloatOp.MOD:
688
+ self.emit_line(f"{dest} = {lhs} {op.op_str[op.op]} {rhs};")
689
+ else:
690
+ # TODO: This may set errno as a side effect, that is a little sketchy.
691
+ self.emit_line(f"{dest} = fmod({lhs}, {rhs});")
692
+
693
+ def visit_float_neg(self, op: FloatNeg) -> None:
694
+ dest = self.reg(op)
695
+ src = self.reg(op.src)
696
+ self.emit_line(f"{dest} = -{src};")
697
+
698
+ def visit_float_comparison_op(self, op: FloatComparisonOp) -> None:
699
+ dest = self.reg(op)
700
+ lhs = self.reg(op.lhs)
701
+ rhs = self.reg(op.rhs)
702
+ self.emit_line(f"{dest} = {lhs} {op.op_str[op.op]} {rhs};")
703
+
704
+ def visit_load_mem(self, op: LoadMem) -> None:
705
+ dest = self.reg(op)
706
+ src = self.reg(op.src)
707
+ # TODO: we shouldn't dereference to type that are pointer type so far
708
+ type = self.ctype(op.type)
709
+ self.emit_line(f"{dest} = *({type} *){src};")
710
+
711
+ def visit_set_mem(self, op: SetMem) -> None:
712
+ dest = self.reg(op.dest)
713
+ src = self.reg(op.src)
714
+ dest_type = self.ctype(op.dest_type)
715
+ # clang whines about self assignment (which we might generate
716
+ # for some casts), so don't emit it.
717
+ if dest != src:
718
+ self.emit_line(f"*({dest_type} *){dest} = {src};")
719
+
720
+ def visit_get_element_ptr(self, op: GetElementPtr) -> None:
721
+ dest = self.reg(op)
722
+ src = self.reg(op.src)
723
+ # TODO: support tuple type
724
+ assert isinstance(op.src_type, RStruct)
725
+ assert op.field in op.src_type.names, "Invalid field name."
726
+ self.emit_line(
727
+ "{} = ({})&(({} *){})->{};".format(
728
+ dest, op.type._ctype, op.src_type.name, src, op.field
729
+ )
730
+ )
731
+
732
+ def visit_load_address(self, op: LoadAddress) -> None:
733
+ typ = op.type
734
+ dest = self.reg(op)
735
+ if isinstance(op.src, Register):
736
+ src = self.reg(op.src)
737
+ elif isinstance(op.src, LoadStatic):
738
+ prefix = self.PREFIX_MAP[op.src.namespace]
739
+ src = self.emitter.static_name(op.src.identifier, op.src.module_name, prefix)
740
+ else:
741
+ src = op.src
742
+ self.emit_line(f"{dest} = ({typ._ctype})&{src};")
743
+
744
+ def visit_keep_alive(self, op: KeepAlive) -> None:
745
+ # This is a no-op.
746
+ pass
747
+
748
+ def visit_unborrow(self, op: Unborrow) -> None:
749
+ # This is a no-op that propagates the source value.
750
+ dest = self.reg(op)
751
+ src = self.reg(op.src)
752
+ self.emit_line(f"{dest} = {src};")
753
+
754
+ # Helpers
755
+
756
+ def label(self, label: BasicBlock) -> str:
757
+ return self.emitter.label(label)
758
+
759
+ def reg(self, reg: Value) -> str:
760
+ if isinstance(reg, Integer):
761
+ val = reg.value
762
+ if val == 0 and is_pointer_rprimitive(reg.type):
763
+ return "NULL"
764
+ s = str(val)
765
+ if val >= (1 << 31):
766
+ # Avoid overflowing signed 32-bit int
767
+ if val >= (1 << 63):
768
+ s += "ULL"
769
+ else:
770
+ s += "LL"
771
+ elif val == -(1 << 63):
772
+ # Avoid overflowing C integer literal
773
+ s = "(-9223372036854775807LL - 1)"
774
+ elif val <= -(1 << 31):
775
+ s += "LL"
776
+ return s
777
+ elif isinstance(reg, Float):
778
+ r = repr(reg.value)
779
+ if r == "inf":
780
+ return "INFINITY"
781
+ elif r == "-inf":
782
+ return "-INFINITY"
783
+ elif r == "nan":
784
+ return "NAN"
785
+ return r
786
+ else:
787
+ return self.emitter.reg(reg)
788
+
789
+ def ctype(self, rtype: RType) -> str:
790
+ return self.emitter.ctype(rtype)
791
+
792
+ def c_error_value(self, rtype: RType) -> str:
793
+ return self.emitter.c_error_value(rtype)
794
+
795
+ def c_undefined_value(self, rtype: RType) -> str:
796
+ return self.emitter.c_undefined_value(rtype)
797
+
798
+ def emit_line(self, line: str, *, ann: object = None) -> None:
799
+ self.emitter.emit_line(line, ann=ann)
800
+
801
+ def emit_lines(self, *lines: str) -> None:
802
+ self.emitter.emit_lines(*lines)
803
+
804
+ def emit_inc_ref(self, dest: str, rtype: RType) -> None:
805
+ self.emitter.emit_inc_ref(dest, rtype, rare=self.rare)
806
+
807
+ def emit_dec_ref(self, dest: str, rtype: RType, is_xdec: bool) -> None:
808
+ self.emitter.emit_dec_ref(dest, rtype, is_xdec=is_xdec, rare=self.rare)
809
+
810
+ def emit_declaration(self, line: str) -> None:
811
+ self.declarations.emit_line(line)
812
+
813
+ def emit_traceback(self, op: Branch) -> None:
814
+ if op.traceback_entry is not None:
815
+ self.emitter.emit_traceback(self.source_path, self.module_name, op.traceback_entry)
816
+
817
+ def emit_attribute_error(self, op: Branch, class_name: str, attr: str) -> None:
818
+ assert op.traceback_entry is not None
819
+ globals_static = self.emitter.static_name("globals", self.module_name)
820
+ self.emit_line(
821
+ 'CPy_AttributeError("%s", "%s", "%s", "%s", %d, %s);'
822
+ % (
823
+ self.source_path.replace("\\", "\\\\"),
824
+ op.traceback_entry[0],
825
+ class_name,
826
+ attr,
827
+ op.traceback_entry[1],
828
+ globals_static,
829
+ )
830
+ )
831
+ if DEBUG_ERRORS:
832
+ self.emit_line('assert(PyErr_Occurred() != NULL && "failure w/o err!");')
833
+
834
+ def emit_signed_int_cast(self, type: RType) -> str:
835
+ if is_tagged(type):
836
+ return "(Py_ssize_t)"
837
+ else:
838
+ return ""
839
+
840
+ def emit_unsigned_int_cast(self, type: RType) -> str:
841
+ if is_int32_rprimitive(type):
842
+ return "(uint32_t)"
843
+ elif is_int64_rprimitive(type):
844
+ return "(uint64_t)"
845
+ else:
846
+ return ""