RandomSec/MUOR/MUOR/signals.py

from django.db.models.signals import post_save
from django.dispatch import receiver
from .DockerManager import DockerManager
from .NGINXConfigurator import NGINXConfigurator
from .models import Profile, Session
from django.contrib.auth.signals import user_logged_in, user_logged_out


def get_sessions_data():
    return tuple((session.profile.volume, session.port, session.sessionid)
                 for session in Session.objects.all())


@receiver(post_save, sender=Profile)
def profile_saved(sender, instance, **kwargs):
    # Check if volume already exists and if not -> creating new volume
    if instance.volume is None:
        docker_manager = DockerManager()
        volume = docker_manager.create_new_volume(instance.username)
        instance.volume = volume

        instance.save()


@receiver(user_logged_in)
def profile_start_up(sender, user, request, **kwargs):
    try:
        user.session = request.session.session_key
        user.session.save()
    except Profile.session.RelatedObjectDoesNotExist:
        # Create session
        session = Session()
        session.user = user
        session.sessionid = request.session.session_key

        # Create container
        docker_manager = DockerManager()

        port = 6969
        container_id = -1
        while True:
            try:
                container_id = docker_manager.turn_on_container(user.volume,
                                                                port)
                break
                # TODO dodać konkret bledy
            except:
                port += 1

        session.port = port
        session.container_id = container_id
        session.save()
    finally:
        NGINXConfigurator.refresh_config(get_sessions_data())


@receiver(user_logged_out)
def profile_logged_out(sender, user, request, **kwargs):
    # Obtaining container_id
    session = Session.objects.get(user=user)
    container_id = session.container_id

    # Turn off and delete container
    docker_manager = DockerManager()
    docker_manager.turn_off_container(container_id)

    # Delete session
    session.delete()
    NGINXConfigurator.refresh_config(get_sessions_data())