fixup! feat(tron): Add support for VoteWitnessContract - core

Author: PrisionMike

core/src/apps/tron/sign_tx.py

@@ -56,7 +56,6 @@ async def sign_tx(msg: TronSignTx, keychain: Keychain) -> TronSignature:
     if fee_limit > _MAX_FEE_LIMIT:
         raise DataError("Tron: fees too high")
 
-    # TODO: CONTRACT_TYPES are all the Tron message types. Can we do better than manually writing them again in consts.py?
     contract = await call_any(messages.TronContractRequest(), *consts.CONTRACT_TYPES)
     raw_contract = await process_contract(contract, fee_limit)
 
@@ -151,9 +150,9 @@ async def process_contract(
         await layout.confirm_withdraw_unfreeze(contract.owner_address)
 
     elif messages.TronVoteWitnessContract.is_type_of(contract):
-        contract_type = TronRawContractType.VoteWitnessContract
         if len(contract.votes) > 9:
             raise DataError("Tron: too many votes")
+        contract_type = TronRawContractType.VoteWitnessContract
         await layout.confirm_votes(contract)
     else:
         raise DataError("Tron: contract type unknown")