#!/usr/bin/env python3 import sys import random import time import contextlib import termios import timeout_decorator import os from blessings import Terminal term = Terminal() height = term.height width = term.width last_dir = 'x' class QuitGameError(BaseException): pass class RanIntoSomethingError(QuitGameError): pass if len(sys.argv) == 2: try: size = int(sys.argv[1]) if size <= 0 or size >= width-13: size = 7 except ValueError: size = 7 else: size = 7 score = 0 def draw_worm(): for location in worm_locations: print(term.move(*reversed(location)) + term.bright_blue('o'), end='') print(term.move(*reversed(worm_head)) + '@' + term.move(*reversed(worm_head)), end='') def draw_frame(): print(term.clear, end='') print(term.move(0, 0) + term.on_red(' worm') + term.bright_cyan_on_red('.py ') + ' Press ' + term.bold_green('I') + ' for info, ' + term.bold_red('Ctrl-C')+ ' to quit', end='') if score > -1: print(term.move(0, width-12) + f'Score:{" "*(4-len(str(score)))}{term.bright_green(str(score))}', end='') print(term.move(1, 0) + term.white_on_red('┌' + ('─' * (width-3)) + '┐'), end='') for y in range(2, height-1): print(term.move(y, 0) + term.white_on_red('│' + term.move(y, width-2) + '│'), end='') print( term.move(height-1, 0) + term.white_on_red('└') + term.white_on_red('─' * (width-3)) + term.white_on_red('┘') , end='') print(term.move(*reversed(bonus_location)) + term.black_on_bright_green(str(bonus_points)), end='') draw_worm() sys.stdout.flush() def move_head(direction, distance): global size, worm_head, last_dir, worm_locations, bonus_location, bonus_points, score if direction not in 'hjklHJKLABCDwasdx': raise ValueError('argument to move_head() must be one of `hjklHJKLABCDwasdx`') if direction == 'x': return worm_locations.append(worm_head.copy()) worm_head = worm_head.copy() if direction in 'kKAw': #up worm_head[1] -= 1 last_dir = 'A' elif direction in 'jJBs': #down worm_head[1] += 1 last_dir = 'B' elif direction in 'hHDa': #left worm_head[0] -= 1 last_dir = 'D' elif direction in 'lLCd': #right worm_head[0] += 1 last_dir = 'C' while len(worm_locations) > size: worm_locations.pop(0) if worm_head in worm_locations \ or worm_head[1] <= 1 \ or worm_head[1] >= height-1 \ or worm_head[0] <= 0 \ or worm_head[0] >= width-2: raise RanIntoSomethingError() if worm_head == bonus_location: size += bonus_points score += bonus_points bonus_points = random.randint(1, 9) while worm_head == bonus_location or bonus_location in worm_locations: bonus_location = [ random.randint(1, width-3), random.randint(2, height-2), ] if distance > 1: move_head(direction, distance-1) @contextlib.contextmanager def decanonize(fd): old_settings = termios.tcgetattr(fd) new_settings = old_settings[:] new_settings[3] &= ~termios.ICANON termios.tcsetattr(fd, termios.TCSAFLUSH, new_settings) yield termios.tcsetattr(fd, termios.TCSAFLUSH, old_settings) worm_y = height // 2 worm_locations = [[i+10, worm_y] for i in range(size)] worm_head = [size+10, worm_y] bonus_points = random.randint(1, 9) bonus_location = worm_head while worm_head == bonus_location or bonus_location in worm_locations: bonus_location = [ random.randint(1, width-3), random.randint(2, height-2), ] do_automove = True do_help = False @timeout_decorator.timeout(1) def run(*_): global do_automove, do_help if do_automove: move_head(last_dir, 1) draw_frame() do_automove = True while True: k = sys.stdin.read(1) if k in 'hjklHJKLABCDwasd': move_head(k, 10 if k in 'HJKL' else 1) draw_frame() do_automove = False return elif k in '': # That string is Ctrl-C Ctrl-\, in case you editor doesn't handle # control characters as well as Vim does. sys.exit(0) elif k in 'iI': do_help = True return try: with term.fullscreen(): os.system('stty raw -echo') draw_frame() while True: if do_help: os.system('stty -raw') print(term.clear() + term.move(0, 0) + term.on_red(' worm') + term.bright_cyan_on_red('.py ') + f''' v1.0: bsdgames worm, ported to Python and improved See https://github.com/kj7rrv/worm.py for source code and installation instructions. Thanks to the authors of the following libraries: * blessings\t\t{term.blue("https://pypi.org/project/blessings/")} * timeout-decorator\t{term.blue("https://pypi.org/project/timeout-decorator/")} Also, thanks to the devolopers of Python and bsdgames worm. It would have been much harder to port worm to Python if either if either worm or Python did not exist. Use the arrow keys or WASD to move. Try to get the green numbers, but don't let the worm run into itself or the red edge. To change the initial length of the worm, add the desired length of the worm after `{term.bright_red('worm')}{term.bright_cyan('.py')}`, as in `{term.bright_red('worm')}{term.bright_cyan('.py')} 20` for a twenty-character-long worm. {term.bright_red('worm')}{term.bright_cyan('.py')} is released under the MIT license.''') print(term.move(height - 1, 0) + 'Press ' + term.bold_green('C') + ' to continue, ' + term.bold_red('Ctrl-C') + ' to exit the game...', end='') sys.stdout.flush() os.system('stty raw') while True: k = sys.stdin.read(1) if k in 'cC': break elif k in '': # That string is Ctrl-C Ctrl-\, in case you editor doesn't handle # control characters as well as Vim does. sys.exit(0) os.system('stty -raw') print(term.clear() + term.move(0, 0) + term.on_red(' worm') + term.bright_cyan_on_red('.py ') + f''' Copyright and License Info Copyright (c) 2021 Samuel L. Sloniker Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. ''' + term.bold('''THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.''')) print(term.move(height - 1, 0) + 'Press ' + term.bold_green('C') + ' to return to the game, ' + term.bold_red('Ctrl-C') + ' to exit...', end='') sys.stdout.flush() os.system('stty raw') while True: k = sys.stdin.read(1) if k in 'cC': break elif k in '': # That string is Ctrl-C Ctrl-\, in case you editor doesn't handle # control characters as well as Vim does. sys.exit(0) do_help = False else: try: run() except timeout_decorator.timeout_decorator.TimeoutError: pass except KeyboardInterrupt: sys.exit(0) except RanIntoSomethingError: os.system('stty -raw echo') print('', end='\r\n') if size + 1 >= (height-3) * (width-1): print('You won!', end='\r\n') else: print('Well, you ran into something and the game is over.', end='\r\n') print(f'Your final score was {score}', end='\r\n') print('', end='\r\n') finally: os.system('stty -raw echo')