A tabbed viewer for Chris Edwards' ASL Vehicle/Gun data cards.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
asl-cards/add_card_dialog.py

98 lines
3.8 KiB

import os
from PyQt5 import uic
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QDialog , QListWidgetItem
import asl_cards.db as db
from constants import *
import globals
# ---------------------------------------------------------------------
class AddCardDialog( QDialog ) :
"""Allow the user to select an ASL card, based on nationality & card type."""
def __init__( self , parent ) :
# initialize
self.selected_card = None
super(AddCardDialog,self).__init__( parent=parent )
# initialize the dialog
uic.loadUi( os.path.join(globals.base_dir,"ui/add_card_dialog.ui") , self )
self.setMinimumSize( self.size() )
self.lb_cards.setSortingEnabled( True )
w = self.buttons_widget
self.xmargin = w.x()
self.ymargin = self.size().height() - (w.y() + w.height())
# load the dialog
for nationality in globals.cards :
self.cbo_nationality.addItem( nationality )
# connect our handlers
self.cbo_nationality.currentIndexChanged[str].connect( self.on_nationality_changed )
self.ok_button.clicked.connect( self.on_ok )
for rb in [self.rb_vehicles,self.rb_ordnance] :
rb.clicked.connect( self.on_card_type_changed )
# select the initial nationality (this will load the rest of the dialog)
self.cbo_nationality.setCurrentIndex( 0 )
self.on_nationality_changed( self.cbo_nationality.itemText(0) )
def on_ok( self ) :
# accept the currently selected card
item = self.lb_cards.currentItem()
self.selected_card = item.data(Qt.UserRole) if item else None
self.accept()
def on_nationality_changed( self , val ) :
"""Update the dialog when the active nationality is changed."""
# reload the available cards for the selected nationality
cards = globals.cards[ val ]
self.lb_cards.clear()
# update the vehicle/ordnance radio boxes
if self.rb_vehicles.isChecked() :
curr_rb = self.rb_vehicles
elif self.rb_ordnance.isChecked() :
curr_rb = self.rb_ordnance
else :
curr_rb = None
self.rb_vehicles.setEnabled( db.TAGTYPE_VEHICLE in cards )
self.rb_ordnance.setEnabled( db.TAGTYPE_ORDNANCE in cards )
if curr_rb is None or not curr_rb.isEnabled() :
for rb in [self.rb_vehicles,self.rb_ordnance] :
if rb.isEnabled() :
rb.setChecked( True )
break
# reload the cards
self.on_card_type_changed()
def on_card_type_changed( self ) :
"""Update the dialog when the active card type is changed."""
self.lb_cards.clear()
# figure out what type of cards to show
if self.rb_vehicles.isChecked() :
card_type = db.TAGTYPE_VEHICLE
elif self.rb_ordnance.isChecked() :
card_type = db.TAGTYPE_ORDNANCE
else :
return
# reload the available cards
cards = globals.cards[ self.cbo_nationality.currentText() ]
cards = cards.get( card_type )
if cards is None :
assert False
return
for card in cards :
item = QListWidgetItem( card.name )
item.setData( Qt.UserRole , card )
self.lb_cards.addItem( item )
self.lb_cards.setCurrentRow( 0 )
self.lb_cards.setFocus()
def resizeEvent( self , evt ) :
# handle the event
w = self.buttons_widget
self.buttons_widget.setGeometry(
self.xmargin , self.size().height() - self.ymargin - w.height() ,
self.size().width() - 2*self.xmargin , w.height()
)
w = self.lb_cards
w.resize( self.size().width() - 2*self.xmargin , self.buttons_widget.y() - w.y() )