Format code with black
This commit is contained in:
parent
39ff3a5672
commit
74035ad320
|
|
@ -29,26 +29,47 @@ class IsFullScreen(BaseException):
|
|||
|
||||
def draw_worm():
|
||||
for location in worm_locations:
|
||||
print(do_move(*reversed(location)) + term.bright_blue('o'), end='')
|
||||
print(do_move(*reversed(location)) + term.bright_blue("o"), end="")
|
||||
|
||||
print(do_move(*reversed(worm_head)) + '@' + do_move(*reversed(worm_head)), end='')
|
||||
print(do_move(*reversed(worm_head)) + "@" + do_move(*reversed(worm_head)), end="")
|
||||
|
||||
|
||||
def draw_frame():
|
||||
print(term.clear, end='')
|
||||
print(do_move(0, 0) + term.on_red(' Worm') + term.black_on_bright_cyan('.py ') + ' Press ' + term.bold_green('I') + ' for info, ' + term.bold_red('Ctrl-C')+ ' to quit', end='')
|
||||
if score > -1:
|
||||
print(do_move(0, width-12) + f'Score:{" "*(4-len(str(score)))}{term.bright_green(str(score))}', end='')
|
||||
print(do_move(1, 0) + term.white_on_red('┌' + ('─' * (width-3)) + '┐'), end='')
|
||||
for y in range(2, height-1):
|
||||
print(do_move(y, 0) + term.white_on_red('│' + do_move(y, width-2) + '│'), end='')
|
||||
print(term.clear, end="")
|
||||
print(
|
||||
do_move(height-1, 0)
|
||||
+ term.white_on_red('└')
|
||||
+ term.white_on_red('─' * (width-3))
|
||||
+ term.white_on_red('┘')
|
||||
, end='')
|
||||
print(do_move(*reversed(bonus_location)) + term.black_on_green(term.on_bright_green(str(bonus_points))), end='')
|
||||
do_move(0, 0)
|
||||
+ term.on_red(" Worm")
|
||||
+ term.black_on_bright_cyan(".py ")
|
||||
+ " Press "
|
||||
+ term.bold_green("I")
|
||||
+ " for info, "
|
||||
+ term.bold_red("Ctrl-C")
|
||||
+ " to quit",
|
||||
end="",
|
||||
)
|
||||
if score > -1:
|
||||
print(
|
||||
do_move(0, width - 12)
|
||||
+ f'Score:{" "*(4-len(str(score)))}{term.bright_green(str(score))}',
|
||||
end="",
|
||||
)
|
||||
print(do_move(1, 0) + term.white_on_red("┌" + ("─" * (width - 3)) + "┐"), end="")
|
||||
for y in range(2, height - 1):
|
||||
print(
|
||||
do_move(y, 0) + term.white_on_red("│" + do_move(y, width - 2) + "│"), end=""
|
||||
)
|
||||
print(
|
||||
do_move(height - 1, 0)
|
||||
+ term.white_on_red("└")
|
||||
+ term.white_on_red("─" * (width - 3))
|
||||
+ term.white_on_red("┘"),
|
||||
end="",
|
||||
)
|
||||
print(
|
||||
do_move(*reversed(bonus_location))
|
||||
+ term.black_on_green(term.on_bright_green(str(bonus_points))),
|
||||
end="",
|
||||
)
|
||||
draw_worm()
|
||||
sys.stdout.flush()
|
||||
|
||||
|
|
@ -56,37 +77,39 @@ def draw_frame():
|
|||
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 not in "hjklHJKLABCDwasdx":
|
||||
raise ValueError("argument to move_head() must be one of `hjklHJKLABCDwasdx`")
|
||||
|
||||
if direction == 'x':
|
||||
if direction == "x":
|
||||
return
|
||||
|
||||
worm_locations.append(worm_head.copy())
|
||||
|
||||
worm_head = worm_head.copy()
|
||||
|
||||
if direction in 'kKAw': #up
|
||||
if direction in "kKAw": # up
|
||||
worm_head[1] -= 1
|
||||
last_dir = 'A'
|
||||
elif direction in 'jJBs': #down
|
||||
last_dir = "A"
|
||||
elif direction in "jJBs": # down
|
||||
worm_head[1] += 1
|
||||
last_dir = 'B'
|
||||
elif direction in 'hHDa': #left
|
||||
last_dir = "B"
|
||||
elif direction in "hHDa": # left
|
||||
worm_head[0] -= 1
|
||||
last_dir = 'D'
|
||||
elif direction in 'lLCd': #right
|
||||
last_dir = "D"
|
||||
elif direction in "lLCd": # right
|
||||
worm_head[0] += 1
|
||||
last_dir = 'C'
|
||||
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:
|
||||
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:
|
||||
|
|
@ -95,7 +118,7 @@ def move_head(direction, distance):
|
|||
new_bonus()
|
||||
|
||||
if distance > 1:
|
||||
move_head(direction, distance-1)
|
||||
move_head(direction, distance - 1)
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
|
|
@ -119,16 +142,16 @@ def run(*_):
|
|||
|
||||
save_game()
|
||||
|
||||
k = await_keys('hjklHJKLABCDwasdiI ')
|
||||
if k in 'hjklHJKLABCDwasd':
|
||||
move_head(k, 10 if k in 'HJKL' else 1)
|
||||
k = await_keys("hjklHJKLABCDwasdiI ")
|
||||
if k in "hjklHJKLABCDwasd":
|
||||
move_head(k, 10 if k in "HJKL" else 1)
|
||||
draw_frame()
|
||||
do_automove = False
|
||||
return
|
||||
elif k in ' ':
|
||||
elif k in " ":
|
||||
save_game()
|
||||
sys.exit(0)
|
||||
elif k in 'iI':
|
||||
elif k in "iI":
|
||||
do_help = True
|
||||
return
|
||||
|
||||
|
|
@ -146,8 +169,8 @@ def new_bonus():
|
|||
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),
|
||||
random.randint(1, width - 3),
|
||||
random.randint(2, height - 2),
|
||||
]
|
||||
|
||||
|
||||
|
|
@ -159,37 +182,65 @@ def do_move(y, x):
|
|||
|
||||
def save_game():
|
||||
if not args.full_screen:
|
||||
with open(gamesave_path, 'w+') as f:
|
||||
json.dump([worm_locations, worm_head, last_dir, score, size, bonus_location, bonus_points, do_automove, do_help], f)
|
||||
with open(gamesave_path, "w+") as f:
|
||||
json.dump(
|
||||
[
|
||||
worm_locations,
|
||||
worm_head,
|
||||
last_dir,
|
||||
score,
|
||||
size,
|
||||
bonus_location,
|
||||
bonus_points,
|
||||
do_automove,
|
||||
do_help,
|
||||
],
|
||||
f,
|
||||
)
|
||||
|
||||
|
||||
term = Terminal()
|
||||
gamesave_path = os.path.join(os.getenv('HOME'), '.worm.py-gamesave')
|
||||
gamesave_path = os.path.join(os.getenv("HOME"), ".worm.py-gamesave")
|
||||
|
||||
parser = argparse.ArgumentParser(prog='wormpy')
|
||||
parser = argparse.ArgumentParser(prog="wormpy")
|
||||
group = parser.add_mutually_exclusive_group()
|
||||
group.add_argument("--save-game", '-s', help="use 80x24 box and save game on ^C (default)", action='store_true', default=True)
|
||||
group.add_argument("--full-screen", '-f', help="use entire screen (disables game saving)", action='store_true')
|
||||
group.add_argument("--delete-save", help="delete saved game and exit", action='store_true')
|
||||
group.add_argument(
|
||||
"--save-game",
|
||||
"-s",
|
||||
help="use 80x24 box and save game on ^C (default)",
|
||||
action="store_true",
|
||||
default=True,
|
||||
)
|
||||
group.add_argument(
|
||||
"--full-screen",
|
||||
"-f",
|
||||
help="use entire screen (disables game saving)",
|
||||
action="store_true",
|
||||
)
|
||||
group.add_argument(
|
||||
"--delete-save", help="delete saved game and exit", action="store_true"
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.delete_save:
|
||||
try:
|
||||
shutil.move(gamesave_path, gamesave_path + '-old')
|
||||
shutil.move(gamesave_path, gamesave_path + "-old")
|
||||
print("Saved game deleted.")
|
||||
except FileNotFoundError:
|
||||
print("No saved game found.")
|
||||
except Exception as e:
|
||||
print("An error occurred.")
|
||||
print(str(type(e)).split("'")[1] + ': ' + str(e))
|
||||
cont = input("Do you want to try to non-recoverably delete the saved game? [Y/n] ")
|
||||
if cont[0] in 'Yy':
|
||||
print(str(type(e)).split("'")[1] + ": " + str(e))
|
||||
cont = input(
|
||||
"Do you want to try to non-recoverably delete the saved game? [Y/n] "
|
||||
)
|
||||
if cont[0] in "Yy":
|
||||
try:
|
||||
os.unlink(gamesave_path)
|
||||
print("Saved game deleted.")
|
||||
except Exception as e:
|
||||
print("An error occurred.")
|
||||
print(str(type(e)).split("'")[1] + ': ' + str(e))
|
||||
print(str(type(e)).split("'")[1] + ": " + str(e))
|
||||
sys.exit(1)
|
||||
else:
|
||||
sys.exit(1)
|
||||
|
|
@ -215,12 +266,22 @@ try:
|
|||
|
||||
with open(gamesave_path) as f:
|
||||
save = json.load(f)
|
||||
worm_locations, worm_head, last_dir, score, size, bonus_location, bonus_points, do_automove, do_help = save
|
||||
(
|
||||
worm_locations,
|
||||
worm_head,
|
||||
last_dir,
|
||||
score,
|
||||
size,
|
||||
bonus_location,
|
||||
bonus_points,
|
||||
do_automove,
|
||||
do_help,
|
||||
) = save
|
||||
except (FileNotFoundError, IsFullScreen):
|
||||
size = 7
|
||||
worm_locations = [[i+10, worm_y] for i in range(size)]
|
||||
worm_head = [size+10, worm_y]
|
||||
last_dir = 'x'
|
||||
worm_locations = [[i + 10, worm_y] for i in range(size)]
|
||||
worm_head = [size + 10, worm_y]
|
||||
last_dir = "x"
|
||||
score = 0
|
||||
new_bonus()
|
||||
do_automove = True
|
||||
|
|
@ -235,12 +296,12 @@ try:
|
|||
if do_help:
|
||||
for info in (info_1, info_2):
|
||||
stty.stty(raw=False)
|
||||
print(info, end='')
|
||||
print(info, end="")
|
||||
sys.stdout.flush()
|
||||
stty.stty(raw=True)
|
||||
|
||||
k = await_keys('cC ')
|
||||
if k in ' ':
|
||||
k = await_keys("cC ")
|
||||
if k in " ":
|
||||
save_game()
|
||||
sys.exit(0)
|
||||
|
||||
|
|
@ -260,12 +321,12 @@ except RanIntoSomethingError:
|
|||
except FileNotFoundError:
|
||||
pass
|
||||
stty.stty(raw=False, echo=True)
|
||||
print('', end='\r\n')
|
||||
if size + 1 >= (height-3) * (width-1):
|
||||
print('You won!', end='\r\n')
|
||||
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')
|
||||
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:
|
||||
stty.stty(raw=False, echo=True)
|
||||
|
|
|
|||
|
|
@ -1,6 +1,10 @@
|
|||
def get_infos(term, do_move, height, width, x_offset):
|
||||
info_1 = term.clear() + do_move(0, 0) + term.on_red(' Worm') \
|
||||
+ term.black_on_bright_cyan('.py ') + f''' v1.0: bsdgames worm, ported \
|
||||
info_1 = (
|
||||
term.clear()
|
||||
+ do_move(0, 0)
|
||||
+ term.on_red(" Worm")
|
||||
+ term.black_on_bright_cyan(".py ")
|
||||
+ f""" v1.0: bsdgames worm, ported \
|
||||
to Python and improved
|
||||
|
||||
See https://github.com/kj7rrv/wormpy for source code and installation
|
||||
|
|
@ -24,15 +28,18 @@ after `{term.bright_red('worm')}{term.bright_cyan('py')}`, as in \
|
|||
-long worm.
|
||||
|
||||
{term.bright_red('Worm')}{term.bright_cyan('.py')} is released under the MIT \
|
||||
license.'''\
|
||||
+ '{}Press {} to continue, {} to exit the game...'.format(
|
||||
do_move(height - 1, 0),
|
||||
term.bold_green('C'),
|
||||
term.bold_red('Ctrl-C')
|
||||
license."""
|
||||
+ "{}Press {} to continue, {} to exit the game...".format(
|
||||
do_move(height - 1, 0), term.bold_green("C"), term.bold_red("Ctrl-C")
|
||||
)
|
||||
)
|
||||
|
||||
info_2 = term.clear() + do_move(0, 0) + term.on_red(' Worm') \
|
||||
+ term.black_on_bright_cyan('.py ') + f''' Copyright and License Info
|
||||
info_2 = (
|
||||
term.clear()
|
||||
+ do_move(0, 0)
|
||||
+ term.on_red(" Worm")
|
||||
+ term.black_on_bright_cyan(".py ")
|
||||
+ f""" Copyright and License Info
|
||||
|
||||
Copyright (c) 2021 Samuel L. Sloniker
|
||||
|
||||
|
|
@ -46,19 +53,21 @@ 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 \
|
||||
"""
|
||||
+ 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.''') \
|
||||
+ '{}Press {} to return to the game, {} to exit...'.format(
|
||||
do_move(height - 1, 0),
|
||||
term.bold_green('C'),
|
||||
term.bold_red('Ctrl-C')
|
||||
SOFTWARE."""
|
||||
)
|
||||
info_1 = info_1.replace('\n', '\n' + term.move_x(x_offset))
|
||||
info_2 = info_2.replace('\n', '\n' + term.move_x(x_offset))
|
||||
+ "{}Press {} to return to the game, {} to exit...".format(
|
||||
do_move(height - 1, 0), term.bold_green("C"), term.bold_red("Ctrl-C")
|
||||
)
|
||||
)
|
||||
info_1 = info_1.replace("\n", "\n" + term.move_x(x_offset))
|
||||
info_2 = info_2.replace("\n", "\n" + term.move_x(x_offset))
|
||||
return info_1, info_2
|
||||
|
|
|
|||
|
|
@ -1,12 +1,13 @@
|
|||
import subprocess
|
||||
|
||||
|
||||
def stty(raw=None, echo=None):
|
||||
args = ['stty']
|
||||
args = ["stty"]
|
||||
|
||||
if raw is not None:
|
||||
args.append('raw' if raw else '-raw')
|
||||
args.append("raw" if raw else "-raw")
|
||||
|
||||
if echo is not None:
|
||||
args.append('echo' if echo else '-echo')
|
||||
args.append("echo" if echo else "-echo")
|
||||
|
||||
subprocess.run(args)
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user