from django.contrib import admin
from django.db import models
from .models import TradeJrnl, LuxOptionsTrade, LuxTrade, LuxOptionPurchase, LuxOptionContract
from utils.widgets import AdminImageWidget, LGEntryForm


@admin.register(TradeJrnl)
class TradingJrnlAdmin(admin.ModelAdmin):
    form = LGEntryForm
    class Media:
        js = ('image-loader.js', 'next-prev-links.js')
        css = {
            "all": ("my_styles.css",)
        }


@admin.register(LuxOptionsTrade)
class LuxOptionsTradeAdmin(admin.ModelAdmin):
    list_display = ('symbol', 'date', 'call_put')
    list_filter = ['call_put']
    class Media:
        js = ('image-loader.js', 'next-prev-links.js')
        css = {
            "all": ("my_styles.css",)
        }

@admin.register(LuxTrade)
class LuxTradeAdmin(admin.ModelAdmin):
    list_display = ('symbol', 'status', 'open_date', 'entry_price', 'stop_price', 'close_price', 'shares', 'amount_invested')
    list_filter = ['status', 'open_date']
    class Media:
        js = ('image-loader.js', 'next-prev-links.js')
        css = {
            "all": ("my_styles.css",)
        }


class LuxOptionContactInline(admin.StackedInline):
    model = LuxOptionContract
    extra = 0
    fieldsets = (
        (None, {
            'fields': (
                'contract_open_price',
                'contract_close_price',
                'strike_price',
                'expiration_date',
                'call_put'
            )
        }),
    )


@admin.register(LuxOptionPurchase)
class LuxOptionPurchaseAdmin(admin.ModelAdmin):
    list_display = ('symbol','status', 'profit_loss')
    inlines = [LuxOptionContactInline,]
    class Media:
        js = ('image-loader.js', 'next-prev-links.js')
        css = {
            "all": ("my_styles.css",)
        }