weapon_generator/CommandHandler.py at master · BeubeuCode/weapon_generator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from Weapon import Weapon

class CommandHandler:

    @staticmethod
    def printHelp():
        print("usage : 1st argument is an integer determining the number of weapons generated, the second argument is a string determining the weapon type : \n")
        print("ASSAULT_RIFLE : generates assault rifle type weapon properties")
        print("SHOTGUN : generates shotgun type weapon properties")
        print("SNIPER_RIFLE : generates sniper rifle type weapon properties")
        exit(0)

    @staticmethod
    def argsHandler(self, arg1=None, arg2=None, w=None):
        if arg1 is None and arg2 is None:
            self.printHelp()

        if arg1 is not None and arg2 is not None:
            w.weapon_generation(arg1, arg2)