git.scottworley.com Git - rfc1751/blob - rfc1751.py

   1 # Copyright (c) 2024 Scott Worley <scottworley@scottworley.com>
   2 #
   3 # Permission to use, copy, modify, and distribute this software for any
   4 # purpose with or without fee is hereby granted, provided that the above
   5 # copyright notice and this permission notice appear in all copies.
   6 #
   7 # THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   8 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   9 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  10 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  11 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  12 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  13 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  14
  15
  16 import sys
  17
  18 from rfc1751wordlist import WORD_LIST, WORD_LIST_SIZE
  19
  20
  21 WORD_LIST_INVERTED = {word: i for (i, word) in enumerate(WORD_LIST)}
  22 assert len(WORD_LIST_INVERTED) == WORD_LIST_SIZE
  23
  24
  25 def encode_actual(x: int) -> list[str]:
  26     return [] if x <= 0 else encode_actual(
  27         x // WORD_LIST_SIZE) + [WORD_LIST[x % WORD_LIST_SIZE]]
  28
  29
  30 def encode(x: int) -> list[str]:
  31     return [WORD_LIST[x]] if x == 0 else encode_actual(x)
  32
  33
  34 def decode(x: list[str]) -> int:
  35     return WORD_LIST_SIZE * decode(x[:-1]) + \
  36         WORD_LIST_INVERTED[x[-1]] if x else 0
  37
  38
  39 def main() -> None:
  40     if sys.argv[1].isnumeric():
  41         print(' '.join(encode(int(sys.argv[1]))))
  42     elif len(sys.argv) == 2:
  43         print(decode(sys.argv[1].split(' ')))
  44     else:
  45         print(decode(sys.argv[1:]))
  46
  47
  48 if __name__ == '__main__':
  49     main()