day05.py

from collections import deque
import re
from src.common.utils import SolverFunctions

title = 'Day 5: Supply Stacks'
parser_method = 'strings_per_line'
handle_data = 'paragraph'


class SolveTheDay(SolverFunctions):
    @staticmethod
    def helper(data):
        crates_data = data[0][:-1]
        operations_data = data[1]
    
        operation_number = 1
        operation = None
        operations = {operation_number: operation}
        crates = {}
    
        for letters_position in crates_data:
            for key, letter in enumerate(range(1, len(letters_position), 4)):
                key += 1
                if letters_position[letter] == " ":
                    continue
                if key not in crates:
                    crates[key] = deque(letters_position[letter])
                else:
                    crates[key].appendleft(letters_position[letter])
    
        for operation in operations_data:
            operations[operation_number] = operation
            operation_number += 1
    
        return crates, operations

    @classmethod
    def level_1(cls, data):
        crates, operations = cls.helper(data)
    
        for operation in operations.values():
            # operation = operation.replace(' ', '').replace('move', '').replace('from', '').replace('to', '')
            # _move, _from, _to = operation.strip()
            _move, _from, _to = map(int, re.findall(r'\d+', operation))
            for counter in range(1, _move+1):
                crates[_to].append(crates[_from].pop())
    
        arranged_crates = dict(sorted(crates.items()))
        return ''.join(crate.pop() for crate in arranged_crates.values())

    @classmethod
    def level_2(cls, data):
        crates, operations = cls.helper(data)

        for operation in operations.values():
            _move, _from, _to = map(int, re.findall(r'\d+', operation))
            temp_movement = deque()
            for counter in range(1, _move + 1):
                temp_movement.extendleft(crates[_from].pop())
            crates[_to].extend(temp_movement)

        arranged_crates = dict(sorted(crates.items()))
        top_crates = ''.join(crate.pop() for crate in arranged_crates.values())
        return top_crates