nus-mtp/cs-modify

'''
    This module contains the handler for web requests pertaining to
    the Delete Module page.
'''


from app import RENDER, set_template
import web
from components import model, session
from components.handlers.outcome import Outcome


class DeleteMod(object):
    '''
        This class handles the display of the Delete Module page
        and the deletion of a module
    '''
    def GET(self):
        '''
            Handles the display of the Delete Module page
        '''
        web.header('X-Frame-Options', 'SAMEORIGIN')
        web.header('X-Content-Type-Options', 'nosniff')
        web.header('X-XSS-Protection', '1')
        if not session.validate_session():
            raise web.seeother('/login')
        else:
            module_infos = model.get_new_modules()
            return RENDER.deleteModule(module_infos)


    def POST(self):
        '''
            Handles the deletion of a module
        '''
        # Verify that module exists
        web.header('X-Frame-Options', 'SAMEORIGIN')
        web.header('X-Content-Type-Options', 'nosniff')
        web.header('X-XSS-Protection', '1')
        input_data = model.validate_input(web.input(), ["code"], show_404=False)
        try:
            module_code = input_data.code
        except AttributeError:
            return Outcome().POST("delete_module", False, None)

        # Verify that module's status is 'New'
        module_infos = model.get_new_modules()
        new_modules = [module_info[0] for module_info in module_infos]
        if module_code not in new_modules:
            return Outcome().POST("delete_module", False, None)

        else:
            outcome = model.delete_module(module_code)
            if outcome is False:
                tenta_mounted_modules = model.get_all_tenta_mounted_modules()
                tenta_mounted_module_codes = [module[0] for module in tenta_mounted_modules]
                if module_code in tenta_mounted_module_codes:
                    return Outcome().POST("delete_module", "has_mounting", module_code)
                else:
                    return Outcome().POST("delete_module", "is_starred", module_code)
            else:
                set_template()   # Refresh list of modules in module search
                return Outcome().POST("delete_module", True, module_code)