webull-openapi-python-sdk 1.0.0__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 (295) hide show
  1. samples/__init__.py +1 -0
  2. samples/data/__init__.py +1 -0
  3. samples/data/data_client.py +57 -0
  4. samples/data/data_streaming_client.py +86 -0
  5. samples/data/data_streaming_client_async.py +101 -0
  6. samples/trade/__init__.py +0 -0
  7. samples/trade/trade_client.py +163 -0
  8. samples/trade/trade_client_v2.py +181 -0
  9. samples/trade/trade_event_client.py +47 -0
  10. webull/__init__.py +1 -0
  11. webull/core/__init__.py +12 -0
  12. webull/core/auth/__init__.py +0 -0
  13. webull/core/auth/algorithm/__init__.py +0 -0
  14. webull/core/auth/algorithm/sha_hmac1.py +65 -0
  15. webull/core/auth/algorithm/sha_hmac256.py +75 -0
  16. webull/core/auth/composer/__init__.py +0 -0
  17. webull/core/auth/composer/default_signature_composer.py +125 -0
  18. webull/core/auth/credentials.py +46 -0
  19. webull/core/auth/signers/__init__.py +0 -0
  20. webull/core/auth/signers/app_key_signer.py +72 -0
  21. webull/core/auth/signers/signer.py +48 -0
  22. webull/core/auth/signers/signer_factory.py +58 -0
  23. webull/core/cache/__init__.py +225 -0
  24. webull/core/client.py +410 -0
  25. webull/core/common/__init__.py +0 -0
  26. webull/core/common/api_type.py +19 -0
  27. webull/core/common/easy_enum.py +35 -0
  28. webull/core/common/region.py +7 -0
  29. webull/core/compat.py +85 -0
  30. webull/core/context/__init__.py +0 -0
  31. webull/core/context/request_context_holder.py +33 -0
  32. webull/core/data/endpoints.json +22 -0
  33. webull/core/data/retry_config.json +15 -0
  34. webull/core/endpoint/__init__.py +8 -0
  35. webull/core/endpoint/chained_endpoint_resolver.py +57 -0
  36. webull/core/endpoint/default_endpoint_resolver.py +60 -0
  37. webull/core/endpoint/local_config_regional_endpoint_resolver.py +77 -0
  38. webull/core/endpoint/resolver_endpoint_request.py +46 -0
  39. webull/core/endpoint/user_customized_endpoint_resolver.py +55 -0
  40. webull/core/exception/__init__.py +0 -0
  41. webull/core/exception/error_code.py +23 -0
  42. webull/core/exception/error_msg.py +21 -0
  43. webull/core/exception/exceptions.py +53 -0
  44. webull/core/headers.py +57 -0
  45. webull/core/http/__init__.py +0 -0
  46. webull/core/http/initializer/__init__.py +0 -0
  47. webull/core/http/initializer/client_initializer.py +79 -0
  48. webull/core/http/initializer/token/__init__.py +0 -0
  49. webull/core/http/initializer/token/bean/__init__.py +0 -0
  50. webull/core/http/initializer/token/bean/access_token.py +40 -0
  51. webull/core/http/initializer/token/bean/check_token_request.py +44 -0
  52. webull/core/http/initializer/token/bean/create_token_request.py +45 -0
  53. webull/core/http/initializer/token/bean/refresh_token_request.py +44 -0
  54. webull/core/http/initializer/token/token_manager.py +208 -0
  55. webull/core/http/initializer/token/token_operation.py +72 -0
  56. webull/core/http/method_type.py +43 -0
  57. webull/core/http/protocol_type.py +43 -0
  58. webull/core/http/request.py +121 -0
  59. webull/core/http/response.py +166 -0
  60. webull/core/request.py +278 -0
  61. webull/core/retry/__init__.py +0 -0
  62. webull/core/retry/backoff_strategy.py +102 -0
  63. webull/core/retry/retry_condition.py +214 -0
  64. webull/core/retry/retry_policy.py +63 -0
  65. webull/core/retry/retry_policy_context.py +51 -0
  66. webull/core/utils/__init__.py +0 -0
  67. webull/core/utils/common.py +62 -0
  68. webull/core/utils/data.py +25 -0
  69. webull/core/utils/desensitize.py +33 -0
  70. webull/core/utils/validation.py +49 -0
  71. webull/core/vendored/__init__.py +0 -0
  72. webull/core/vendored/requests/__init__.py +94 -0
  73. webull/core/vendored/requests/__version__.py +28 -0
  74. webull/core/vendored/requests/_internal_utils.py +56 -0
  75. webull/core/vendored/requests/adapters.py +539 -0
  76. webull/core/vendored/requests/api.py +166 -0
  77. webull/core/vendored/requests/auth.py +307 -0
  78. webull/core/vendored/requests/certs.py +34 -0
  79. webull/core/vendored/requests/compat.py +85 -0
  80. webull/core/vendored/requests/cookies.py +555 -0
  81. webull/core/vendored/requests/exceptions.py +136 -0
  82. webull/core/vendored/requests/help.py +134 -0
  83. webull/core/vendored/requests/hooks.py +48 -0
  84. webull/core/vendored/requests/models.py +960 -0
  85. webull/core/vendored/requests/packages/__init__.py +17 -0
  86. webull/core/vendored/requests/packages/certifi/__init__.py +17 -0
  87. webull/core/vendored/requests/packages/certifi/__main__.py +16 -0
  88. webull/core/vendored/requests/packages/certifi/cacert.pem +4433 -0
  89. webull/core/vendored/requests/packages/certifi/core.py +51 -0
  90. webull/core/vendored/requests/packages/chardet/__init__.py +53 -0
  91. webull/core/vendored/requests/packages/chardet/big5freq.py +400 -0
  92. webull/core/vendored/requests/packages/chardet/big5prober.py +61 -0
  93. webull/core/vendored/requests/packages/chardet/chardistribution.py +247 -0
  94. webull/core/vendored/requests/packages/chardet/charsetgroupprober.py +120 -0
  95. webull/core/vendored/requests/packages/chardet/charsetprober.py +159 -0
  96. webull/core/vendored/requests/packages/chardet/cli/__init__.py +1 -0
  97. webull/core/vendored/requests/packages/chardet/cli/chardetect.py +99 -0
  98. webull/core/vendored/requests/packages/chardet/codingstatemachine.py +102 -0
  99. webull/core/vendored/requests/packages/chardet/compat.py +48 -0
  100. webull/core/vendored/requests/packages/chardet/cp949prober.py +63 -0
  101. webull/core/vendored/requests/packages/chardet/enums.py +90 -0
  102. webull/core/vendored/requests/packages/chardet/escprober.py +115 -0
  103. webull/core/vendored/requests/packages/chardet/escsm.py +260 -0
  104. webull/core/vendored/requests/packages/chardet/eucjpprober.py +106 -0
  105. webull/core/vendored/requests/packages/chardet/euckrfreq.py +209 -0
  106. webull/core/vendored/requests/packages/chardet/euckrprober.py +61 -0
  107. webull/core/vendored/requests/packages/chardet/euctwfreq.py +401 -0
  108. webull/core/vendored/requests/packages/chardet/euctwprober.py +60 -0
  109. webull/core/vendored/requests/packages/chardet/gb2312freq.py +297 -0
  110. webull/core/vendored/requests/packages/chardet/gb2312prober.py +60 -0
  111. webull/core/vendored/requests/packages/chardet/hebrewprober.py +306 -0
  112. webull/core/vendored/requests/packages/chardet/jisfreq.py +339 -0
  113. webull/core/vendored/requests/packages/chardet/jpcntx.py +247 -0
  114. webull/core/vendored/requests/packages/chardet/langbulgarianmodel.py +242 -0
  115. webull/core/vendored/requests/packages/chardet/langcyrillicmodel.py +347 -0
  116. webull/core/vendored/requests/packages/chardet/langgreekmodel.py +239 -0
  117. webull/core/vendored/requests/packages/chardet/langhebrewmodel.py +214 -0
  118. webull/core/vendored/requests/packages/chardet/langhungarianmodel.py +239 -0
  119. webull/core/vendored/requests/packages/chardet/langthaimodel.py +213 -0
  120. webull/core/vendored/requests/packages/chardet/langturkishmodel.py +207 -0
  121. webull/core/vendored/requests/packages/chardet/latin1prober.py +159 -0
  122. webull/core/vendored/requests/packages/chardet/mbcharsetprober.py +105 -0
  123. webull/core/vendored/requests/packages/chardet/mbcsgroupprober.py +68 -0
  124. webull/core/vendored/requests/packages/chardet/mbcssm.py +586 -0
  125. webull/core/vendored/requests/packages/chardet/sbcharsetprober.py +146 -0
  126. webull/core/vendored/requests/packages/chardet/sbcsgroupprober.py +87 -0
  127. webull/core/vendored/requests/packages/chardet/sjisprober.py +106 -0
  128. webull/core/vendored/requests/packages/chardet/universaldetector.py +300 -0
  129. webull/core/vendored/requests/packages/chardet/utf8prober.py +96 -0
  130. webull/core/vendored/requests/packages/chardet/version.py +23 -0
  131. webull/core/vendored/requests/packages/urllib3/__init__.py +114 -0
  132. webull/core/vendored/requests/packages/urllib3/_collections.py +346 -0
  133. webull/core/vendored/requests/packages/urllib3/connection.py +405 -0
  134. webull/core/vendored/requests/packages/urllib3/connectionpool.py +910 -0
  135. webull/core/vendored/requests/packages/urllib3/contrib/__init__.py +0 -0
  136. webull/core/vendored/requests/packages/urllib3/contrib/_appengine_environ.py +44 -0
  137. webull/core/vendored/requests/packages/urllib3/contrib/_securetransport/__init__.py +0 -0
  138. webull/core/vendored/requests/packages/urllib3/contrib/_securetransport/bindings.py +607 -0
  139. webull/core/vendored/requests/packages/urllib3/contrib/_securetransport/low_level.py +360 -0
  140. webull/core/vendored/requests/packages/urllib3/contrib/appengine.py +303 -0
  141. webull/core/vendored/requests/packages/urllib3/contrib/ntlmpool.py +125 -0
  142. webull/core/vendored/requests/packages/urllib3/contrib/pyopenssl.py +484 -0
  143. webull/core/vendored/requests/packages/urllib3/contrib/securetransport.py +818 -0
  144. webull/core/vendored/requests/packages/urllib3/contrib/socks.py +206 -0
  145. webull/core/vendored/requests/packages/urllib3/exceptions.py +260 -0
  146. webull/core/vendored/requests/packages/urllib3/fields.py +192 -0
  147. webull/core/vendored/requests/packages/urllib3/filepost.py +112 -0
  148. webull/core/vendored/requests/packages/urllib3/packages/__init__.py +19 -0
  149. webull/core/vendored/requests/packages/urllib3/packages/backports/__init__.py +0 -0
  150. webull/core/vendored/requests/packages/urllib3/packages/backports/makefile.py +67 -0
  151. webull/core/vendored/requests/packages/urllib3/packages/ordered_dict.py +273 -0
  152. webull/core/vendored/requests/packages/urllib3/packages/six.py +882 -0
  153. webull/core/vendored/requests/packages/urllib3/packages/socks.py +887 -0
  154. webull/core/vendored/requests/packages/urllib3/packages/ssl_match_hostname/__init__.py +19 -0
  155. webull/core/vendored/requests/packages/urllib3/packages/ssl_match_hostname/_implementation.py +170 -0
  156. webull/core/vendored/requests/packages/urllib3/poolmanager.py +467 -0
  157. webull/core/vendored/requests/packages/urllib3/request.py +164 -0
  158. webull/core/vendored/requests/packages/urllib3/response.py +721 -0
  159. webull/core/vendored/requests/packages/urllib3/util/__init__.py +68 -0
  160. webull/core/vendored/requests/packages/urllib3/util/connection.py +148 -0
  161. webull/core/vendored/requests/packages/urllib3/util/queue.py +35 -0
  162. webull/core/vendored/requests/packages/urllib3/util/request.py +132 -0
  163. webull/core/vendored/requests/packages/urllib3/util/response.py +101 -0
  164. webull/core/vendored/requests/packages/urllib3/util/retry.py +426 -0
  165. webull/core/vendored/requests/packages/urllib3/util/selectors.py +601 -0
  166. webull/core/vendored/requests/packages/urllib3/util/ssl_.py +396 -0
  167. webull/core/vendored/requests/packages/urllib3/util/timeout.py +256 -0
  168. webull/core/vendored/requests/packages/urllib3/util/url.py +252 -0
  169. webull/core/vendored/requests/packages/urllib3/util/wait.py +164 -0
  170. webull/core/vendored/requests/packages.py +28 -0
  171. webull/core/vendored/requests/sessions.py +750 -0
  172. webull/core/vendored/requests/status_codes.py +105 -0
  173. webull/core/vendored/requests/structures.py +119 -0
  174. webull/core/vendored/requests/utils.py +916 -0
  175. webull/core/vendored/six.py +905 -0
  176. webull/data/__init__.py +3 -0
  177. webull/data/common/__init__.py +0 -0
  178. webull/data/common/category.py +26 -0
  179. webull/data/common/connect_ack.py +29 -0
  180. webull/data/common/direction.py +25 -0
  181. webull/data/common/exchange_code.py +33 -0
  182. webull/data/common/exercise_style.py +22 -0
  183. webull/data/common/expiration_cycle.py +26 -0
  184. webull/data/common/instrument_status.py +23 -0
  185. webull/data/common/option_type.py +20 -0
  186. webull/data/common/subscribe_type.py +22 -0
  187. webull/data/common/timespan.py +29 -0
  188. webull/data/data_client.py +35 -0
  189. webull/data/data_streaming_client.py +89 -0
  190. webull/data/internal/__init__.py +0 -0
  191. webull/data/internal/default_retry_policy.py +84 -0
  192. webull/data/internal/exceptions.py +60 -0
  193. webull/data/internal/quotes_client.py +314 -0
  194. webull/data/internal/quotes_decoder.py +40 -0
  195. webull/data/internal/quotes_payload_decoder.py +35 -0
  196. webull/data/internal/quotes_topic.py +36 -0
  197. webull/data/quotes/__init__.py +0 -0
  198. webull/data/quotes/instrument.py +33 -0
  199. webull/data/quotes/market_data.py +187 -0
  200. webull/data/quotes/market_streaming_data.py +66 -0
  201. webull/data/quotes/subscribe/__init__.py +0 -0
  202. webull/data/quotes/subscribe/ask_bid_result.py +49 -0
  203. webull/data/quotes/subscribe/basic_result.py +45 -0
  204. webull/data/quotes/subscribe/broker_result.py +33 -0
  205. webull/data/quotes/subscribe/message_pb2.py +37 -0
  206. webull/data/quotes/subscribe/order_result.py +30 -0
  207. webull/data/quotes/subscribe/payload_type.py +19 -0
  208. webull/data/quotes/subscribe/quote_decoder.py +28 -0
  209. webull/data/quotes/subscribe/quote_result.py +47 -0
  210. webull/data/quotes/subscribe/snapshot_decoder.py +30 -0
  211. webull/data/quotes/subscribe/snapshot_result.py +69 -0
  212. webull/data/quotes/subscribe/tick_decoder.py +29 -0
  213. webull/data/quotes/subscribe/tick_result.py +47 -0
  214. webull/data/request/__init__.py +0 -0
  215. webull/data/request/get_batch_historical_bars_request.py +43 -0
  216. webull/data/request/get_corp_action_request.py +47 -0
  217. webull/data/request/get_eod_bars_request.py +32 -0
  218. webull/data/request/get_historical_bars_request.py +43 -0
  219. webull/data/request/get_instruments_request.py +30 -0
  220. webull/data/request/get_quotes_request.py +35 -0
  221. webull/data/request/get_snapshot_request.py +38 -0
  222. webull/data/request/get_tick_request.py +37 -0
  223. webull/data/request/subscribe_request.py +43 -0
  224. webull/data/request/unsubscribe_request.py +42 -0
  225. webull/trade/__init__.py +2 -0
  226. webull/trade/common/__init__.py +0 -0
  227. webull/trade/common/account_type.py +22 -0
  228. webull/trade/common/category.py +29 -0
  229. webull/trade/common/combo_ticker_type.py +23 -0
  230. webull/trade/common/combo_type.py +31 -0
  231. webull/trade/common/currency.py +24 -0
  232. webull/trade/common/forbid_reason.py +27 -0
  233. webull/trade/common/instrument_type.py +27 -0
  234. webull/trade/common/markets.py +27 -0
  235. webull/trade/common/order_entrust_type.py +21 -0
  236. webull/trade/common/order_side.py +23 -0
  237. webull/trade/common/order_status.py +25 -0
  238. webull/trade/common/order_tif.py +24 -0
  239. webull/trade/common/order_type.py +30 -0
  240. webull/trade/common/trade_policy.py +22 -0
  241. webull/trade/common/trading_date_type.py +24 -0
  242. webull/trade/common/trailing_type.py +23 -0
  243. webull/trade/events/__init__.py +0 -0
  244. webull/trade/events/default_retry_policy.py +64 -0
  245. webull/trade/events/events_pb2.py +43 -0
  246. webull/trade/events/events_pb2_grpc.py +66 -0
  247. webull/trade/events/signature_composer.py +61 -0
  248. webull/trade/events/types.py +21 -0
  249. webull/trade/request/__init__.py +0 -0
  250. webull/trade/request/cancel_order_request.py +28 -0
  251. webull/trade/request/get_account_balance_request.py +28 -0
  252. webull/trade/request/get_account_positions_request.py +30 -0
  253. webull/trade/request/get_account_profile_request.py +26 -0
  254. webull/trade/request/get_app_subscriptions.py +28 -0
  255. webull/trade/request/get_open_orders_request.py +30 -0
  256. webull/trade/request/get_order_detail_request.py +27 -0
  257. webull/trade/request/get_today_orders_request.py +31 -0
  258. webull/trade/request/get_trade_calendar_request.py +30 -0
  259. webull/trade/request/get_trade_instrument_detail_request.py +24 -0
  260. webull/trade/request/get_trade_security_detail_request.py +42 -0
  261. webull/trade/request/get_tradeable_instruments_request.py +27 -0
  262. webull/trade/request/palce_order_request.py +91 -0
  263. webull/trade/request/place_order_request_v2.py +58 -0
  264. webull/trade/request/replace_order_request.py +73 -0
  265. webull/trade/request/replace_order_request_v2.py +38 -0
  266. webull/trade/request/v2/__init__.py +0 -0
  267. webull/trade/request/v2/cancel_option_request.py +28 -0
  268. webull/trade/request/v2/cancel_order_request.py +28 -0
  269. webull/trade/request/v2/get_account_balance_request.py +28 -0
  270. webull/trade/request/v2/get_account_list.py +23 -0
  271. webull/trade/request/v2/get_account_positions_request.py +24 -0
  272. webull/trade/request/v2/get_order_detail_request.py +26 -0
  273. webull/trade/request/v2/get_order_history_request.py +35 -0
  274. webull/trade/request/v2/palce_order_request.py +87 -0
  275. webull/trade/request/v2/place_option_request.py +64 -0
  276. webull/trade/request/v2/preview_option_request.py +28 -0
  277. webull/trade/request/v2/preview_order_request.py +59 -0
  278. webull/trade/request/v2/replace_option_request.py +28 -0
  279. webull/trade/request/v2/replace_order_request.py +57 -0
  280. webull/trade/trade/__init__.py +0 -0
  281. webull/trade/trade/account_info.py +83 -0
  282. webull/trade/trade/order_operation.py +246 -0
  283. webull/trade/trade/trade_calendar.py +37 -0
  284. webull/trade/trade/trade_instrument.py +72 -0
  285. webull/trade/trade/v2/__init__.py +0 -0
  286. webull/trade/trade/v2/account_info_v2.py +55 -0
  287. webull/trade/trade/v2/order_operation_v2.py +206 -0
  288. webull/trade/trade_client.py +43 -0
  289. webull/trade/trade_events_client.py +233 -0
  290. webull_openapi_python_sdk-1.0.0.dist-info/METADATA +28 -0
  291. webull_openapi_python_sdk-1.0.0.dist-info/RECORD +295 -0
  292. webull_openapi_python_sdk-1.0.0.dist-info/WHEEL +5 -0
  293. webull_openapi_python_sdk-1.0.0.dist-info/licenses/LICENSE +202 -0
  294. webull_openapi_python_sdk-1.0.0.dist-info/licenses/NOTICE +56 -0
  295. webull_openapi_python_sdk-1.0.0.dist-info/top_level.txt +2 -0
@@ -0,0 +1,252 @@
1
+ # Copyright 2022 Webull
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ from __future__ import absolute_import
16
+ from collections import namedtuple
17
+ import re
18
+
19
+ from ..exceptions import LocationParseError
20
+ from ..packages.six.moves.urllib.parse import quote
21
+
22
+
23
+ url_attrs = ['scheme', 'auth', 'host', 'port', 'path', 'query', 'fragment']
24
+
25
+ # We only want to normalize urls with an HTTP(S) scheme.
26
+ # urllib3 infers URLs without a scheme (None) to be http.
27
+ NORMALIZABLE_SCHEMES = ('http', 'https', None)
28
+
29
+ _contains_disallowed_url_pchar_re = re.compile('[\x00-\x20\x7f]')
30
+
31
+
32
+ class Url(namedtuple('Url', url_attrs)):
33
+ """
34
+ Datastructure for representing an HTTP URL. Used as a return value for
35
+ :func:`parse_url`. Both the scheme and host are normalized as they are
36
+ both case-insensitive according to RFC 3986.
37
+ """
38
+ __slots__ = ()
39
+
40
+ def __new__(cls, scheme=None, auth=None, host=None, port=None, path=None,
41
+ query=None, fragment=None):
42
+ if path and not path.startswith('/'):
43
+ path = '/' + path
44
+ if scheme:
45
+ scheme = scheme.lower()
46
+ if host and scheme in NORMALIZABLE_SCHEMES:
47
+ host = host.lower()
48
+ return super(Url, cls).__new__(cls, scheme, auth, host, port, path,
49
+ query, fragment)
50
+
51
+ @property
52
+ def hostname(self):
53
+ """For backwards-compatibility with urlparse. We're nice like that."""
54
+ return self.host
55
+
56
+ @property
57
+ def request_uri(self):
58
+ """Absolute path including the query string."""
59
+ uri = self.path or '/'
60
+
61
+ if self.query is not None:
62
+ uri += '?' + self.query
63
+
64
+ return uri
65
+
66
+ @property
67
+ def netloc(self):
68
+ """Network location including host and port"""
69
+ if self.port:
70
+ return '%s:%d' % (self.host, self.port)
71
+ return self.host
72
+
73
+ @property
74
+ def url(self):
75
+ """
76
+ Convert self into a url
77
+
78
+ This function should more or less round-trip with :func:`.parse_url`. The
79
+ returned url may not be exactly the same as the url inputted to
80
+ :func:`.parse_url`, but it should be equivalent by the RFC (e.g., urls
81
+ with a blank port will have : removed).
82
+
83
+ Example: ::
84
+
85
+ >>> U = parse_url('http://google.com/mail/')
86
+ >>> U.url
87
+ 'http://google.com/mail/'
88
+ >>> Url('http', 'username:password', 'host.com', 80,
89
+ ... '/path', 'query', 'fragment').url
90
+ 'http://username:password@host.com:80/path?query#fragment'
91
+ """
92
+ scheme, auth, host, port, path, query, fragment = self
93
+ url = ''
94
+
95
+ # We use "is not None" we want things to happen with empty strings (or 0 port)
96
+ if scheme is not None:
97
+ url += scheme + '://'
98
+ if auth is not None:
99
+ url += auth + '@'
100
+ if host is not None:
101
+ url += host
102
+ if port is not None:
103
+ url += ':' + str(port)
104
+ if path is not None:
105
+ url += path
106
+ if query is not None:
107
+ url += '?' + query
108
+ if fragment is not None:
109
+ url += '#' + fragment
110
+
111
+ return url
112
+
113
+ def __str__(self):
114
+ return self.url
115
+
116
+
117
+ def split_first(s, delims):
118
+ """
119
+ Given a string and an iterable of delimiters, split on the first found
120
+ delimiter. Return two split parts and the matched delimiter.
121
+
122
+ If not found, then the first part is the full input string.
123
+
124
+ Example::
125
+
126
+ >>> split_first('foo/bar?baz', '?/=')
127
+ ('foo', 'bar?baz', '/')
128
+ >>> split_first('foo/bar?baz', '123')
129
+ ('foo/bar?baz', '', None)
130
+
131
+ Scales linearly with number of delims. Not ideal for large number of delims.
132
+ """
133
+ min_idx = None
134
+ min_delim = None
135
+ for d in delims:
136
+ idx = s.find(d)
137
+ if idx < 0:
138
+ continue
139
+
140
+ if min_idx is None or idx < min_idx:
141
+ min_idx = idx
142
+ min_delim = d
143
+
144
+ if min_idx is None or min_idx < 0:
145
+ return s, '', None
146
+
147
+ return s[:min_idx], s[min_idx + 1:], min_delim
148
+
149
+
150
+ def parse_url(url):
151
+ """
152
+ Given a url, return a parsed :class:`.Url` namedtuple. Best-effort is
153
+ performed to parse incomplete urls. Fields not provided will be None.
154
+
155
+ Partly backwards-compatible with :mod:`urlparse`.
156
+
157
+ Example::
158
+
159
+ >>> parse_url('http://google.com/mail/')
160
+ Url(scheme='http', host='google.com', port=None, path='/mail/', ...)
161
+ >>> parse_url('google.com:80')
162
+ Url(scheme=None, host='google.com', port=80, path=None, ...)
163
+ >>> parse_url('/foo?bar')
164
+ Url(scheme=None, host=None, port=None, path='/foo', query='bar', ...)
165
+ """
166
+
167
+ # While this code has overlap with stdlib's urlparse, it is much
168
+ # simplified for our needs and less annoying.
169
+ # Additionally, this implementations does silly things to be optimal
170
+ # on CPython.
171
+
172
+ if not url:
173
+ # Empty
174
+ return Url()
175
+
176
+ # Prevent CVE-2019-9740.
177
+ # adapted from https://github.com/python/cpython/pull/12755
178
+ url = _contains_disallowed_url_pchar_re.sub(lambda match: quote(match.group()), url)
179
+
180
+ scheme = None
181
+ auth = None
182
+ host = None
183
+ port = None
184
+ path = None
185
+ fragment = None
186
+ query = None
187
+
188
+ # Scheme
189
+ if '://' in url:
190
+ scheme, url = url.split('://', 1)
191
+
192
+ # Find the earliest Authority Terminator
193
+ # (http://tools.ietf.org/html/rfc3986#section-3.2)
194
+ url, path_, delim = split_first(url, ['/', '?', '#'])
195
+
196
+ if delim:
197
+ # Reassemble the path
198
+ path = delim + path_
199
+
200
+ # Auth
201
+ if '@' in url:
202
+ # Last '@' denotes end of auth part
203
+ auth, url = url.rsplit('@', 1)
204
+
205
+ # IPv6
206
+ if url and url[0] == '[':
207
+ host, url = url.split(']', 1)
208
+ host += ']'
209
+
210
+ # Port
211
+ if ':' in url:
212
+ _host, port = url.split(':', 1)
213
+
214
+ if not host:
215
+ host = _host
216
+
217
+ if port:
218
+ # If given, ports must be integers. No whitespace, no plus or
219
+ # minus prefixes, no non-integer digits such as ^2 (superscript).
220
+ if not port.isdigit():
221
+ raise LocationParseError(url)
222
+ try:
223
+ port = int(port)
224
+ except ValueError:
225
+ raise LocationParseError(url)
226
+ else:
227
+ # Blank ports are cool, too. (rfc3986#section-3.2.3)
228
+ port = None
229
+
230
+ elif not host and url:
231
+ host = url
232
+
233
+ if not path:
234
+ return Url(scheme, auth, host, port, path, query, fragment)
235
+
236
+ # Fragment
237
+ if '#' in path:
238
+ path, fragment = path.split('#', 1)
239
+
240
+ # Query
241
+ if '?' in path:
242
+ path, query = path.split('?', 1)
243
+
244
+ return Url(scheme, auth, host, port, path, query, fragment)
245
+
246
+
247
+ def get_host(url):
248
+ """
249
+ Deprecated. Use :func:`parse_url` instead.
250
+ """
251
+ p = parse_url(url)
252
+ return p.scheme or 'http', p.hostname, p.port
@@ -0,0 +1,164 @@
1
+ # Copyright 2022 Webull
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ import errno
16
+ from functools import partial
17
+ import select
18
+ import sys
19
+ try:
20
+ from time import monotonic
21
+ except ImportError:
22
+ from time import time as monotonic
23
+
24
+ __all__ = ["NoWayToWaitForSocketError", "wait_for_read", "wait_for_write"]
25
+
26
+
27
+ class NoWayToWaitForSocketError(Exception):
28
+ pass
29
+
30
+
31
+ # How should we wait on sockets?
32
+ #
33
+ # There are two types of APIs you can use for waiting on sockets: the fancy
34
+ # modern stateful APIs like epoll/kqueue, and the older stateless APIs like
35
+ # select/poll. The stateful APIs are more efficient when you have a lots of
36
+ # sockets to keep track of, because you can set them up once and then use them
37
+ # lots of times. But we only ever want to wait on a single socket at a time
38
+ # and don't want to keep track of state, so the stateless APIs are actually
39
+ # more efficient. So we want to use select() or poll().
40
+ #
41
+ # Now, how do we choose between select() and poll()? On traditional Unixes,
42
+ # select() has a strange calling convention that makes it slow, or fail
43
+ # altogether, for high-numbered file descriptors. The point of poll() is to fix
44
+ # that, so on Unixes, we prefer poll().
45
+ #
46
+ # On Windows, there is no poll() (or at least Python doesn't provide a wrapper
47
+ # for it), but that's OK, because on Windows, select() doesn't have this
48
+ # strange calling convention; plain select() works fine.
49
+ #
50
+ # So: on Windows we use select(), and everywhere else we use poll(). We also
51
+ # fall back to select() in case poll() is somehow broken or missing.
52
+
53
+ if sys.version_info >= (3, 5):
54
+ # Modern Python, that retries syscalls by default
55
+ def _retry_on_intr(fn, timeout):
56
+ return fn(timeout)
57
+ else:
58
+ # Old and broken Pythons.
59
+ def _retry_on_intr(fn, timeout):
60
+ if timeout is None:
61
+ deadline = float("inf")
62
+ else:
63
+ deadline = monotonic() + timeout
64
+
65
+ while True:
66
+ try:
67
+ return fn(timeout)
68
+ # OSError for 3 <= pyver < 3.5, select.error for pyver <= 2.7
69
+ except (OSError, select.error) as e:
70
+ # 'e.args[0]' incantation works for both OSError and select.error
71
+ if e.args[0] != errno.EINTR:
72
+ raise
73
+ else:
74
+ timeout = deadline - monotonic()
75
+ if timeout < 0:
76
+ timeout = 0
77
+ if timeout == float("inf"):
78
+ timeout = None
79
+ continue
80
+
81
+
82
+ def select_wait_for_socket(sock, read=False, write=False, timeout=None):
83
+ if not read and not write:
84
+ raise RuntimeError("must specify at least one of read=True, write=True")
85
+ rcheck = []
86
+ wcheck = []
87
+ if read:
88
+ rcheck.append(sock)
89
+ if write:
90
+ wcheck.append(sock)
91
+ # When doing a non-blocking connect, most systems signal success by
92
+ # marking the socket writable. Windows, though, signals success by marked
93
+ # it as "exceptional". We paper over the difference by checking the write
94
+ # sockets for both conditions. (The stdlib selectors module does the same
95
+ # thing.)
96
+ fn = partial(select.select, rcheck, wcheck, wcheck)
97
+ rready, wready, xready = _retry_on_intr(fn, timeout)
98
+ return bool(rready or wready or xready)
99
+
100
+
101
+ def poll_wait_for_socket(sock, read=False, write=False, timeout=None):
102
+ if not read and not write:
103
+ raise RuntimeError("must specify at least one of read=True, write=True")
104
+ mask = 0
105
+ if read:
106
+ mask |= select.POLLIN
107
+ if write:
108
+ mask |= select.POLLOUT
109
+ poll_obj = select.poll()
110
+ poll_obj.register(sock, mask)
111
+
112
+ # For some reason, poll() takes timeout in milliseconds
113
+ def do_poll(t):
114
+ if t is not None:
115
+ t *= 1000
116
+ return poll_obj.poll(t)
117
+
118
+ return bool(_retry_on_intr(do_poll, timeout))
119
+
120
+
121
+ def null_wait_for_socket(*args, **kwargs):
122
+ raise NoWayToWaitForSocketError("no select-equivalent available")
123
+
124
+
125
+ def _have_working_poll():
126
+ # Apparently some systems have a select.poll that fails as soon as you try
127
+ # to use it, either due to strange configuration or broken monkeypatching
128
+ # from libraries like eventlet/greenlet.
129
+ try:
130
+ poll_obj = select.poll()
131
+ _retry_on_intr(poll_obj.poll, 0)
132
+ except (AttributeError, OSError):
133
+ return False
134
+ else:
135
+ return True
136
+
137
+
138
+ def wait_for_socket(*args, **kwargs):
139
+ # We delay choosing which implementation to use until the first time we're
140
+ # called. We could do it at import time, but then we might make the wrong
141
+ # decision if someone goes wild with monkeypatching select.poll after
142
+ # we're imported.
143
+ global wait_for_socket
144
+ if _have_working_poll():
145
+ wait_for_socket = poll_wait_for_socket
146
+ elif hasattr(select, "select"):
147
+ wait_for_socket = select_wait_for_socket
148
+ else: # Platform-specific: Appengine.
149
+ wait_for_socket = null_wait_for_socket
150
+ return wait_for_socket(*args, **kwargs)
151
+
152
+
153
+ def wait_for_read(sock, timeout=None):
154
+ """ Waits for reading to be available on a given socket.
155
+ Returns True if the socket is readable, or False if the timeout expired.
156
+ """
157
+ return wait_for_socket(sock, read=True, timeout=timeout)
158
+
159
+
160
+ def wait_for_write(sock, timeout=None):
161
+ """ Waits for writing to be available on a given socket.
162
+ Returns True if the socket is readable, or False if the timeout expired.
163
+ """
164
+ return wait_for_socket(sock, write=True, timeout=timeout)
@@ -0,0 +1,28 @@
1
+ # Copyright 2022 Webull
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ import sys
16
+
17
+ # This code exists for backwards compatibility reasons.
18
+ # I don't like it either. Just look the other way. :)
19
+
20
+ for package in ('urllib3', 'idna', 'chardet'):
21
+ locals()[package] = __import__(package)
22
+ # This traversal is apparently necessary such that the identities are
23
+ # preserved (requests.packages.urllib3.* is urllib3.*)
24
+ for mod in list(sys.modules):
25
+ if mod == package or mod.startswith(package + '.'):
26
+ sys.modules['requests.packages.' + mod] = sys.modules[mod]
27
+
28
+ # Kinda cool, though, right?