command_handler_proxy_server.py

import cmd
from io import StringIO

from services.user_service import UserService
from consts import OUT_OF_NETWORK_ERROR

class ClientCommandHandler(cmd.Cmd):
    INVALID_ARGS = f'invalid arguments'
    prompt = '(youtube) '
    user_service = UserService()

    def has_logged_in(self):
        return self.user_service.user != None

    def do_help(self, arg: str) -> str:
        'help or ?'
        help_output = StringIO()
        super().__setattr__('stdout', help_output)
        super().do_help(arg)
        return help_output.getvalue(), False

    def do_login(self, arg):
        'login [type= admin | superadmin] [username] [password]'
        args = parse(arg)
        if len(args) < 3:
            return ClientCommandHandler.INVALID_ARGS, False
        elif args[0] == 'admin':
            return self.user_service.get_admin(username=args[1], password=args[2]), False
        elif args[0] == 'superadmin':
            return self.user_service.get_super_admin(username=args[1], password=args[2]), False
        else:
            return ClientCommandHandler.INVALID_ARGS, False

    def do_logout(self, arg):
        'logout'
        self.user_service.logout()
        return f'logout successfull.', False
    
    def do_whoami(self, arg):
        'whoami'
        user = self.user_service.user
        return user.username if user else 'No one is logged in', False

    def do_signup(self, arg):
        'signup [username] [password]'
        args = parse(arg)
        if len(args) < 2:
            return ClientCommandHandler.INVALID_ARGS, False
        self.user_service.create_admin(args[0], args[1])
        return f'signup successfull. Super admin permission is needed.', False

    def do_exit(self, arg):
        'type q to exit'
        return '', False    

    def forward_req(self):
        if not self.has_logged_in():
            return OUT_OF_NETWORK_ERROR, False
        return self.user_service.user.username, True
    
    def do_permissions(self, arg):
        'permissions'
        return self.forward_req()
    
    def do_approve(self, arg):
        'approve [username]'
        return self.forward_req()

    def do_ticket(self, arg):
        'ticket [text]'
        return self.forward_req()

    def do_reply_ticket(self, arg):
        'reply_ticket [ticketid] [text]'
        return self.forward_req()

    def do_set_ticket_state(self, arg):
        'set_ticket_state [ticketid] [state]'
        return self.forward_req()

    def do_open_tickets(self, arg):
        'open_tickets'
        return self.forward_req()

    def do_label(self, arg):
        'label [video_title] [text]'
        return self.forward_req()

    def do_remove(self, arg):
        'remove [video_title]'
        return self.forward_req()
    
    def do_unstrike(self, arg):
        'unstrike [username]'
        return self.forward_req()


def parse(arg):
    return arg.split()