from django.contrib import admin
from django.db import models
from .models import TradeJrnl, OptionsTrade, 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(OptionsTrade)
class OptionsTradeAdmin(admin.ModelAdmin):
    list_display = ('symbol', 'date', 'buy_sell', 'open_close', 'quantity')
    list_filter = ['buy_sell']
    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",)
        }