from django.contrib import admin
from django.db import models
from .models import TradeJrnl, LuxOptionsTrade, LuxTrade
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",)
        }