Архивы Python

Опрос клавиатуры при помощи curses

17.04.2024 Павел Грибов Python, Из жизни

Опрос клавиатуры при помощи curses несколько запутан. Во первых есть способ получить одиночное нажатие кнопки при помощи:

res=stdscr.getkey()

1	res=stdscr.getkey()

В этом случае в res попадёт непосредственно сам символ в виде строки.

Второй способ — использование :

res=stdscr.getch()

1	res=stdscr.getch()

В этом случае в res попадет число вида int — код нажатой кнопки. Однако оба способа не работают, когда нужно получить данные о нажатой специальной клавиши, типа стрелок, F1..F12 и т.д. Если необходимо их получить, то нужно сначала включить данную возможность:

stdscr.keypad(True)

1	stdscr.keypad(True)

Зачем так сделано, загадка. Почему нельзя получить эти данные сразу?

Так-же есть способ отключить ожидание нажатия кнопки, при помощи:

curses.cbreak()

1	curses.cbreak()

Итог: задача «Опрос клавиатуры при помощи curses» выполнена. Можете почитать и другие заметки посвященные Python. Официальная документация на curses здесь

Оставить комментарий curses, клавиатура, опрос клавиатуры

Квест в консоли на Python. Часть 2

15.04.2024 Павел Грибов Python

Итак, продолжаем продолжаем писать квест в консоли на языке Python. Первая часть описана здесь. В ней мы реализовали автоматическую загрузку и сохранение состояния прохождения квеста. Сейчас же займемся (начнем по крайне мере) отрисовкой локации, и реакцией на нажатые кнопки. В локации предусмотрим возможность отображения картинки из ASCII. Например json стартовой локации может выглядеть примерно так:

{
  "title": "Стартовая локация",
  "description": "Стартовая локация описание",
  "ascii_art": "                    __            ================================\n         ALCATRAZ  /__\\            ||     ||<(.)>||<(.)>||     || \n       ____________|  |            ||    _||     ||     ||_    || \n       |_|_|_|_|_|_|  |            ||   (__D     ||     C__)   || \n       |_|_|_|_|_|_|__|            ||   (__D     ||     C__)   ||\n      A@\\|_|_|_|_|_|/@@Aa          ||   (__D     ||     C__)   ||\n   aaA@@@@@@@@@@@@@@@@@@@aaaA      ||   (__D     ||     C__)   ||\n  A@@@@@@@@@@@DWB@@@@@@@@@@@@A     ||     ||     ||     ||  dwb||\n^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^  ================================",
  "available_locations": {
    "left": 1,
    "right": 2,
    "forward": 3,
    "back": 4
  }
}

{

"title": "Стартовая локация",

"description": "Стартовая локация описание",

"ascii_art": " __ ================================\n ALCATRAZ /__\\ || ||<(.)>||<(.)>|| || \n ____________| | || _|| || ||_ || \n |_|_|_|_|_|_| | || (__D || C__) || \n |_|_|_|_|_|_|__| || (__D || C__) ||\n A@\\|_|_|_|_|_|/@@Aa || (__D || C__) ||\n aaA@@@@@@@@@@@@@@@@@@@aaaA || (__D || C__) ||\n A@@@@@@@@@@@DWB@@@@@@@@@@@@A || || || || dwb||\n^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ================================",

"available_locations": {

"left": 1,

"right": 2,

"forward": 3,

"back": 4

}

Создадим класс TLocation, при инициалиизации будем передавать в него инициализированный класс player. В переменной класса data — будем хранить загруженную локацию.

class TLocation:
    data = {}
    player={}
    stdscr=curses.initscr()
    scr_size=stdscr.getmaxyx()
    def __init__(self,player):
        self.player=player;
    def load_location(self, location):
        f = open("locations/" + str(location) + ".json", mode='r', encoding='utf-8')
        self.data = json.load(f)
        f.close()

class TLocation:

data = {}

player={}

stdscr=curses.initscr()

scr_size=stdscr.getmaxyx()

def __init__(self,player):

self.player=player;

def load_location(self, location):

f = open("locations/" + str(location) + ".json", mode='r', encoding='utf-8')

self.data = json.load(f)

f.close()

Далее нарисуем верхнее меню, где сообщаем игроку, на какие локации он может перемещаться, и что он держит в руках:

    def top_menu(self):
        loc="Идти: "
        if "left" in self.data["available_locations"]:
            loc=loc+"влево(4) "
        if "right" in self.data["available_locations"]:
            loc=loc+"вправо(6) "
        if "forward" in self.data["available_locations"]:
            loc=loc+"вперед(8) "
        if "back" in self.data["available_locations"]:
            loc=loc+"назад(2) "
        curses.start_color()
        curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)
        self.stdscr.addstr(1, 1, loc,curses.color_pair(1))
        curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)
        self.stdscr.addstr(1, 1, loc,curses.color_pair(1))
        # что в руках
        hands = "В руках: "
        self.stdscr.addstr(2, 1, "")
        for object in self.player.data["in_hands"]:
            hands=hands+"["+object+"]"
        if len(self.player.data["in_hands"])==0:
            hands = hands+"ничего нет"
        self.stdscr.addstr(2, 1, hands, curses.color_pair(1))

def top_menu(self):

loc="Идти: "

if "left" in self.data["available_locations"]:

loc=loc+"влево(4) "

if "right" in self.data["available_locations"]:

loc=loc+"вправо(6) "

if "forward" in self.data["available_locations"]:

loc=loc+"вперед(8) "

if "back" in self.data["available_locations"]:

loc=loc+"назад(2) "

curses.start_color()

curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)

self.stdscr.addstr(1, 1, loc,curses.color_pair(1))

curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)

self.stdscr.addstr(1, 1, loc,curses.color_pair(1))

# что в руках

hands = "В руках: "

self.stdscr.addstr(2, 1, "")

for object in self.player.data["in_hands"]:

hands=hands+"["+object+"]"

if len(self.player.data["in_hands"])==0:

hands = hands+"ничего нет"

self.stdscr.addstr(2, 1, hands, curses.color_pair(1))

Ну и собственно основной код отрисовки локации, включающий бесконечный цикл ожидания нажатий клавиатуры. Предусматриваем переход на другую локацию, выход из игры и сохранение игры.

    def location_view(self, location):
        self.load_location(location)
        self.stdscr.clear()
        self.stdscr.border()
        self.top_menu()  # рисуем верхнее меню
        # название локации
        curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)
        x = int((self.scr_size[1] - len(self.data["title"])) / 2)
        self.stdscr.addstr(2, x, self.data["title"],curses.color_pair(2))
        # рисуем рисунок ежели он есть
        y=3
        if "ascii_art" in self.data:
            mass_art=self.data["ascii_art"].split("\n")
            i=0
            while i<len(mass_art):
                x=int((self.scr_size[1]-len(mass_art[i]))/2)
                self.stdscr.addstr(y+i, x, mass_art[i])
                i+=1
            y = y + len(mass_art);
         # Выводим описательную часть
        curses.init_pair(3, curses.COLOR_CYAN, curses.COLOR_BLACK)
        self.stdscr.addstr(y, 1, self.data["description"],curses.color_pair(3))
        # Выводим нижнее меню
        curses.init_pair(4, curses.COLOR_RED, curses.COLOR_BLACK)
        self.stdscr.addstr(y + 1, 1, "Осмотреться вокруг [v] Применить что в руках [h]", curses.color_pair(4))
        self.stdscr.addstr(y + 2, 1, "Выйти из квеста [q] Сохранить состояние [r]", curses.color_pair(4))
        self.stdscr.refresh()
        while True:
            key=self.stdscr.getch()
            print(key)
            # реализация перехода с локации на локацию
            if key==52 and "left" in self.data["available_locations"]:
                self.player.data["location"]=self.data["available_locations"]["left"]
                self.location_view(self.player.data["location"])
            if key==54 and "right" in self.data["available_locations"]:
                self.player.data["location"]=self.data["available_locations"]["right"]
                self.location_view(self.player.data["location"])
            if key==56 and "forward" in self.data["available_locations"]:
                self.player.data["location"]=self.data["available_locations"]["forward"]
                self.location_view(self.player.data["location"])
            if key==50 and "back" in self.data["available_locations"]:
                self.player.data["location"]=self.data["available_locations"]["back"]
                self.location_view(self.player.data["location"])
            if key == 114:
                self.player.save()
                #self.message("Внимание!","Состояние прохождения завершено. \nФайл находится в папке /saves")
            if key==113:
                curses.reset_shell_mode()
                curses.endwin()
                exit(0)
        print(self.data)

def location_view(self, location):

self.load_location(location)

self.stdscr.clear()

self.stdscr.border()

self.top_menu() # рисуем верхнее меню

# название локации

curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)

x = int((self.scr_size[1] - len(self.data["title"])) / 2)

self.stdscr.addstr(2, x, self.data["title"],curses.color_pair(2))

# рисуем рисунок ежели он есть

y=3

if "ascii_art" in self.data:

mass_art=self.data["ascii_art"].split("\n")

i=0

while i<len(mass_art):

x=int((self.scr_size[1]-len(mass_art[i]))/2)

self.stdscr.addstr(y+i, x, mass_art[i])

i+=1

y = y + len(mass_art);

# Выводим описательную часть

curses.init_pair(3, curses.COLOR_CYAN, curses.COLOR_BLACK)

self.stdscr.addstr(y, 1, self.data["description"],curses.color_pair(3))

# Выводим нижнее меню

curses.init_pair(4, curses.COLOR_RED, curses.COLOR_BLACK)

self.stdscr.addstr(y + 1, 1, "Осмотреться вокруг [v] Применить что в руках [h]", curses.color_pair(4))

self.stdscr.addstr(y + 2, 1, "Выйти из квеста [q] Сохранить состояние [r]", curses.color_pair(4))

self.stdscr.refresh()

while True:

key=self.stdscr.getch()

print(key)

# реализация перехода с локации на локацию

if key==52 and "left" in self.data["available_locations"]:

self.player.data["location"]=self.data["available_locations"]["left"]

self.location_view(self.player.data["location"])

if key==54 and "right" in self.data["available_locations"]:

self.player.data["location"]=self.data["available_locations"]["right"]

self.location_view(self.player.data["location"])

if key==56 and "forward" in self.data["available_locations"]:

self.player.data["location"]=self.data["available_locations"]["forward"]

self.location_view(self.player.data["location"])

if key==50 and "back" in self.data["available_locations"]:

self.player.data["location"]=self.data["available_locations"]["back"]

self.location_view(self.player.data["location"])

if key == 114:

self.player.save()

#self.message("Внимание!","Состояние прохождения завершено. \nФайл находится в папке /saves")

if key==113:

curses.reset_shell_mode()

curses.endwin()

exit(0)

print(self.data)

В результате картинка (квест в консоли) на мониторе выглядит уже чуть симпатичнее:

Оставить комментарий python, движек, квест, консоль

Квест в консоли на Python. Часть 1

11.04.2024 Павел Грибов Python

Дело было вечером, делать было нечего (с). Ну не то чтобы совсем нечего, но выдалась свободное немножко время, поэтому для того чтобы не забыть (да уж чего там, и вспомнить уже) окончательно Python, решил сделать маленький движёк для текстовых квестов с выполняющихся в консоли (квест в консоли).

Сначала определимся что где и как:

Локации будем описывать в формате json
Локации будем складывать в папку locations. Имена файлов — номер локации.
В ходе квеста можно «сохраняться», чтобы была возможность продолжить квест
Сохранения будем хранить в папке saves
Все классы храним в папке classes

В результате у меня получилась такая структура папок и файлов:

Первым делом нарисую минимальный json стартовой локации:

{
  "title": "Стартовая страница",
  "description": "Стартовая страница описание",
  "available_locations": {
    "left": 1,
    "right": 2,
    "forward": 3,
    "back": 4
  }
}

{

"title": "Стартовая страница",

"description": "Стартовая страница описание",

"available_locations": {

"left": 1,

"right": 2,

"forward": 3,

"back": 4

}

Т.е. начинаем на стартовой локации (0), доступны переходы в локации 1,2,3 и 4.

Далее реализуем класс игрока, с реализацией функционала сохранения и стадии прохождения квеста:

from datetime import datetime as dt
import json
import os


class TPlayer:
    data = {}

    def __init__(self, name):
        self.data["name"] = name  # Имя пользователя
        self.data["location"] = 0  # текущая локация

    def load(self, filename):
        """
        Загрузить состояние квеста из файла
        :param filename: имя файла из папки saves
        """
        f = open("saves/" + filename, mode='r', encoding='utf-8')
        self.data = json.load(f)
        f.close()

    def list_saves(self):
        """
            Показать доступные сохранения
        """
        files = os.listdir("saves")
        print(files)

    def save(self):
        """
            Сохранить текущее состояние пользователя
        """
        time = dt.now()
        filename = time.strftime("%d_%m_%Y_%H_%M") + ".save"
        print(filename)
        f = open("saves/" + filename, mode='w', encoding='utf-8')
        json.dump(self.data, f)
        f.close()

from datetime import datetime as dt

import json

import os

class TPlayer:

data = {}

def __init__(self, name):

self.data["name"] = name # Имя пользователя

self.data["location"] = 0 # текущая локация

def load(self, filename):

"""

Загрузить состояние квеста из файла

:param filename: имя файла из папки saves

"""

f = open("saves/" + filename, mode='r', encoding='utf-8')

self.data = json.load(f)

f.close()

def list_saves(self):

"""

Показать доступные сохранения

"""

files = os.listdir("saves")

print(files)

def save(self):

"""

Сохранить текущее состояние пользователя

"""

time = dt.now()

filename = time.strftime("%d_%m_%Y_%H_%M") + ".save"

print(filename)

f = open("saves/" + filename, mode='w', encoding='utf-8')

json.dump(self.data, f)

f.close()

В главном файле (main.py), реализуем проверку аргументов командной строки и переход к началу квеста:

#!/usr/bin/env python3
# encoding: utf-8
import classes.player as tplayer
import classes.location as tlocation
import sys

player = tplayer.TPlayer("Васян")
location = tlocation.TLocation()


def start_location():
    print(player.data["name"])
    location.location_view(player.data["location"])

if __name__ == '__main__':
    for param in sys.argv:
        if param == "--load":
            if len(sys.argv) == 2:
                print("Ошибка: нет имени файла")
                exit(0)
            player.load(sys.argv[2])
            start_location()
            exit(0)
        if param == "--new":
            exit(0)
        if param == "--list":
            player.list_saves()
            exit(0)
    if len(sys.argv) == 1:
        print("Для запуска квеста необходимо использовать следующие параметры:")
        print("  --load   - загрузить сохранение и начать квест")
        print("  --list   - получить список сохранений")
        print("  --new <имя участника>    - начать квест заново")
        exit()

#!/usr/bin/env python3

# encoding: utf-8

import classes.player as tplayer

import classes.location as tlocation

import sys

player = tplayer.TPlayer("Васян")

location = tlocation.TLocation()

def start_location():

print(player.data["name"])

location.location_view(player.data["location"])

if __name__ == '__main__':

for param in sys.argv:

if param == "--load":

if len(sys.argv) == 2:

print("Ошибка: нет имени файла")

exit(0)

player.load(sys.argv[2])

start_location()

exit(0)

if param == "--new":

exit(0)

if param == "--list":

player.list_saves()

exit(0)

if len(sys.argv) == 1:

print("Для запуска квеста необходимо использовать следующие параметры:")

print(" --load - загрузить сохранение и начать квест")

print(" --list - получить список сохранений")

print(" --new <имя участника> - начать квест заново")

exit()

Разработка «квест в консоли» может быть действительно просто.. Вы можете посмотреть и другие мои статьи посвященные разработке на Python

Один комментарий python, квест в консоли

Python: ошибка установки cv2

20.11.2023 Павел Грибов Python

При попытке установить:

pip install cv2

1	pip install cv2

Получаем ошибку:

ERROR: Could not find a version that satisfies the requirement cv2 (from versions: none)

1	ERROR: Could not find a version that satisfies the requirement cv2 (from versions: none)

А всё просто. Почему то авторы пакета обозвали его в репозитарии pip как opencv-python. Соответственно и устанавливать его нужно так:

pip install opencv-python

1	pip install opencv-python

А использовать так:

import cv2

1	import cv2

Оставить комментарий cv2

Сохранение весов модели нейросети

16.11.2023 Павел Грибов Python

В продолжении статьи Создаём нейросеть на Python, Итак сеть создали, натренировали. А что делать чтобы решение можно было определять по картинке лето или зима, не обучая каждый раз модель заново? Ну так ведь можно просто сохранить полученные веса нейросети в файл, А затем уже их загружать в случае необходимости.

В Tensotflow уже есть встроенный функционал для сохранения весов в файл. Для этого используется функция вызова кэлбека после прохождения каждого шага обучения. В моём случае промежуточные модели обучения не нужны, потому параметр save_freq делаю равным количеству итераций обучения:

traning_model_save=base_dir+"\save_model\cp.ckpt"
....
EPOCHS = 10
print("- настраиваем кэлбеки для сохранения натренированной модели")
# Создаем колбек для сохранения контрольной точки
cp_callback = tf.keras.callbacks.ModelCheckpoint(traning_model_save,
                                                 save_weights_only=True,
                                                 save_freq=EPOCHS,
                                                 verbose=1)
print("- тренируем модель")
history = model.fit(
    train_data_gen,
    steps_per_epoch=int(np.ceil(total_train / float(BATCH_SIZE))),
    epochs=EPOCHS,
    validation_data=val_data_gen,
    validation_steps=int(np.ceil(total_val / float(BATCH_SIZE))),
    callbacks = [cp_callback]   # вызываем кэлбек после каждого шага обучения
)

traning_model_save=base_dir+"\save_model\cp.ckpt"

....

EPOCHS = 10

print("- настраиваем кэлбеки для сохранения натренированной модели")

# Создаем колбек для сохранения контрольной точки

cp_callback = tf.keras.callbacks.ModelCheckpoint(traning_model_save,

save_weights_only=True,

save_freq=EPOCHS,

verbose=1)

print("- тренируем модель")

history = model.fit(

train_data_gen,

steps_per_epoch=int(np.ceil(total_train / float(BATCH_SIZE))),

epochs=EPOCHS,

validation_data=val_data_gen,

validation_steps=int(np.ceil(total_val / float(BATCH_SIZE))),

callbacks = [cp_callback] # вызываем кэлбек после каждого шага обучения

)

Так-же можно сохранять не только веса, но и всю модель целиком:

model.save('model.h5')

1	model.save('model.h5')

Оставить комментарий callbacks, python, нейросеть, сохранение весов

1 2 3 … 12 »

Пн	Вт	Ср	Чт	Пт	Сб	Вс
				1	2	3
4	5	6	7	8	9	10
11	12	13	14	15	16	17
18	19	20	21	22	23	24
25	26	27	28	29	30