1 files changed, 15 insertions, 8 deletions
diff --git a/app/trading/forms.py b/app/trading/forms.py
index 60c6de6..cd2128d 100644
--- a/app/trading/forms.py
+++ b/app/trading/forms.py
@@ -1,13 +1,13 @@
 from django import forms
 from django.forms.utils import ValidationError
 
-from .models import LuxOptionContact
+from .models import LuxOptionContract
 
 class LuxOptionsForm(forms.ModelForm):
     contracts = forms.IntegerField()
 
     class Meta:
-        model = LuxOptionContact
+        model = LuxOptionContract
         fields = [
             'symbol', 
             'strike_price',
@@ -17,9 +17,16 @@ class LuxOptionsForm(forms.ModelForm):
             'contracts'
         ]
 
-    def save(self, commit=True):
-        i = 0
-        while i < int(self.cleaned_data['contracts']):
-            print(i, "contracts")
-            i = i+1
-        return super(LuxOptionsForm, self).save(commit=commit)
+
+class LuxOptionsUpdateForm(forms.ModelForm):
+    contracts_to_close = forms.IntegerField()
+
+    class Meta:
+        model = LuxOptionContract
+        fields = [
+            'symbol', 
+            'strike_price',
+            'expiration_date',
+            'contracts_to_close',
+            'contract_close_price',
+        ]