Skip to content

Commit 407a464

Browse files
morningveramorningvera
authored
add contract type to models (#236)
1 parent aefe876 commit 407a464

File tree

2 files changed

+23
-0
lines changed

2 files changed

+23
-0
lines changed

‎polygon/rest/models/contracts.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ class OptionsContract:
1919
"OptionsContract contains data for a specified ticker symbol."
2020
additional_underlyings: Optional[List[Underlying]] = None
2121
cfi: Optional[str] = None
22+
contract_type: Optional[str] = None
2223
correction: Optional[str] = None
2324
exercise_style: Optional[str] = None
2425
expiration_date: Optional[str] = None
@@ -35,6 +36,7 @@ def from_dict(d):
3536
if "additional_underlyings" not in d
3637
else [Underlying.from_dict(u) for u in d["additional_underlyings"]],
3738
cfi=d.get("cfi", None),
39+
contract_type=d.get("contract_type", None),
3840
correction=d.get("correction", None),
3941
exercise_style=d.get("exercise_style", None),
4042
expiration_date=d.get("expiration_date", None),

‎test_rest/test_contracts.py‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ def test_get_options_contract(self):
88
expected = OptionsContract(
99
additional_underlyings=None,
1010
cfi="OCASPS",
11+
contract_type="call",
1112
correction=None,
1213
exercise_style="american",
1314
expiration_date="2024-01-19",
@@ -26,6 +27,7 @@ def test_list_options_contracts(self):
2627
OptionsContract(
2728
additional_underlyings=None,
2829
cfi="OCASPS",
30+
contract_type="call",
2931
correction=None,
3032
exercise_style="american",
3133
expiration_date="2022-05-20",
@@ -39,6 +41,7 @@ def test_list_options_contracts(self):
3941
OptionsContract(
4042
additional_underlyings=None,
4143
cfi="OCASPS",
44+
contract_type="call",
4245
correction=None,
4346
exercise_style="american",
4447
expiration_date="2022-05-20",
@@ -52,6 +55,7 @@ def test_list_options_contracts(self):
5255
OptionsContract(
5356
additional_underlyings=None,
5457
cfi="OCASPS",
58+
contract_type="call",
5559
correction=None,
5660
exercise_style="american",
5761
expiration_date="2022-05-20",
@@ -65,6 +69,7 @@ def test_list_options_contracts(self):
6569
OptionsContract(
6670
additional_underlyings=None,
6771
cfi="OCASPS",
72+
contract_type="call",
6873
correction=None,
6974
exercise_style="american",
7075
expiration_date="2022-05-20",
@@ -78,6 +83,7 @@ def test_list_options_contracts(self):
7883
OptionsContract(
7984
additional_underlyings=None,
8085
cfi="OCASPS",
86+
contract_type="call",
8187
correction=None,
8288
exercise_style="american",
8389
expiration_date="2022-05-20",
@@ -91,6 +97,7 @@ def test_list_options_contracts(self):
9197
OptionsContract(
9298
additional_underlyings=None,
9399
cfi="OCASPS",
100+
contract_type="call",
94101
correction=None,
95102
exercise_style="american",
96103
expiration_date="2022-05-20",
@@ -104,6 +111,7 @@ def test_list_options_contracts(self):
104111
OptionsContract(
105112
additional_underlyings=None,
106113
cfi="OCASPS",
114+
contract_type="call",
107115
correction=None,
108116
exercise_style="american",
109117
expiration_date="2022-05-20",
@@ -117,6 +125,7 @@ def test_list_options_contracts(self):
117125
OptionsContract(
118126
additional_underlyings=None,
119127
cfi="OCASPS",
128+
contract_type="call",
120129
correction=None,
121130
exercise_style="american",
122131
expiration_date="2022-05-20",
@@ -130,6 +139,7 @@ def test_list_options_contracts(self):
130139
OptionsContract(
131140
additional_underlyings=None,
132141
cfi="OCASPS",
142+
contract_type="call",
133143
correction=None,
134144
exercise_style="american",
135145
expiration_date="2022-05-20",
@@ -143,6 +153,7 @@ def test_list_options_contracts(self):
143153
OptionsContract(
144154
additional_underlyings=None,
145155
cfi="OCASPS",
156+
contract_type="call",
146157
correction=None,
147158
exercise_style="american",
148159
expiration_date="2022-05-20",
@@ -156,6 +167,7 @@ def test_list_options_contracts(self):
156167
OptionsContract(
157168
additional_underlyings=None,
158169
cfi="OCASPS",
170+
contract_type="call",
159171
correction=None,
160172
exercise_style="american",
161173
expiration_date="2022-05-20",
@@ -169,6 +181,7 @@ def test_list_options_contracts(self):
169181
OptionsContract(
170182
additional_underlyings=None,
171183
cfi="OCASPS",
184+
contract_type="call",
172185
correction=None,
173186
exercise_style="american",
174187
expiration_date="2022-05-20",
@@ -182,6 +195,7 @@ def test_list_options_contracts(self):
182195
OptionsContract(
183196
additional_underlyings=None,
184197
cfi="OCASPS",
198+
contract_type="call",
185199
correction=None,
186200
exercise_style="american",
187201
expiration_date="2022-05-20",
@@ -195,6 +209,7 @@ def test_list_options_contracts(self):
195209
OptionsContract(
196210
additional_underlyings=None,
197211
cfi="OCASPS",
212+
contract_type="call",
198213
correction=None,
199214
exercise_style="american",
200215
expiration_date="2022-05-20",
@@ -208,6 +223,7 @@ def test_list_options_contracts(self):
208223
OptionsContract(
209224
additional_underlyings=None,
210225
cfi="OCASPS",
226+
contract_type="call",
211227
correction=None,
212228
exercise_style="american",
213229
expiration_date="2022-05-20",
@@ -221,6 +237,7 @@ def test_list_options_contracts(self):
221237
OptionsContract(
222238
additional_underlyings=None,
223239
cfi="OCASPS",
240+
contract_type="call",
224241
correction=None,
225242
exercise_style="american",
226243
expiration_date="2022-05-20",
@@ -234,6 +251,7 @@ def test_list_options_contracts(self):
234251
OptionsContract(
235252
additional_underlyings=None,
236253
cfi="OCASPS",
254+
contract_type="call",
237255
correction=None,
238256
exercise_style="american",
239257
expiration_date="2022-05-20",
@@ -247,6 +265,7 @@ def test_list_options_contracts(self):
247265
OptionsContract(
248266
additional_underlyings=None,
249267
cfi="OCASPS",
268+
contract_type="call",
250269
correction=None,
251270
exercise_style="american",
252271
expiration_date="2022-05-20",
@@ -260,6 +279,7 @@ def test_list_options_contracts(self):
260279
OptionsContract(
261280
additional_underlyings=None,
262281
cfi="OCASPS",
282+
contract_type="call",
263283
correction=None,
264284
exercise_style="american",
265285
expiration_date="2022-05-20",
@@ -273,6 +293,7 @@ def test_list_options_contracts(self):
273293
OptionsContract(
274294
additional_underlyings=None,
275295
cfi="OCASPS",
296+
contract_type="call",
276297
correction=None,
277298
exercise_style="american",
278299
expiration_date="2022-05-20",

0 commit comments

Comments
 (0)