'''--- Day 2: Bathroom Security ---

You arrive at Easter Bunny Headquarters under cover of darkness. However, you left in such a rush that you forgot to use the bathroom! Fancy office buildings like this one usually have keypad locks on their bathrooms, so you search the front desk for the code.

"In order to improve security," the document you find says, "bathroom codes will no longer be written down. Instead, please memorize and follow the procedure below to access the bathrooms."

The document goes on to explain that each button to be pressed can be found by starting on the previous button and moving to adjacent buttons on the keypad: U moves up, D moves down, L moves left, and R moves right. Each line of instructions corresponds to one button, starting at the previous button (or, for the first line, the "5" button); press whatever button you're on at the end of each line. If a move doesn't lead to a button, ignore it.

You can't hold it much longer, so you decide to figure out the code as you walk to the bathroom. You picture a keypad like this:

1 2 3
4 5 6
7 8 9

Suppose your instructions are:

ULL
RRDDD
LURDL
UUUUD

    You start at "5" and move up (to "2"), left (to "1"), and left (you can't, and stay on "1"), so the first button is 1.
    Starting from the previous button ("1"), you move right twice (to "3") and then down three times (stopping at "9" after two moves and ignoring the third), ending up with 9.
    Continuing from "9", you move left, up, right, down, and left, ending with 8.
    Finally, you move up four times (stopping at "2"), then down once, ending with 5.

So, in this example, the bathroom code is 1985.

Your puzzle input is the instructions from the document you found at the front desk. What is the bathroom code?
'''

if __name__ == '__main__':
  with open("day02input.txt") as fp:
    directions = fp.readlines()

    moves = {
              1:{'L':1, 'R':2, 'U':1, 'D':4},
              2:{'L':1, 'R':3, 'U':2, 'D':5},
              3:{'L':2, 'R':3, 'U':3, 'D':6},
              4:{'L':4, 'R':5, 'U':1, 'D':7},
              5:{'L':4, 'R':6, 'U':2, 'D':8},
              6:{'L':5, 'R':6, 'U':3, 'D':9},
              7:{'L':7, 'R':8, 'U':4, 'D':7},
              8:{'L':7, 'R':9, 'U':5, 'D':8},
              9:{'L':8, 'R':9, 'U':6, 'D':9}
            }

    code = [5 for x in range(len(directions))]

    for index, digit in enumerate(code):
      for instruction in directions[index]:
        if instruction != '\n':
          digit = moves[digit][instruction]
      code[index] = digit

    print(code)






