Можно ли через python создать и сохранить файл в формате png или jpg?

Question

[YukiLina]

Я написала програму, что-то похожее на paint и хочу, чтобы при нажатии на кнопку сохранялся рисунок в памяти компьютера. Подскажите пожалуйста как это сделать.

Comments

[Модератор]

YukiLina дорогой пользователь, настоятельно рекомендуем еще раз обратить самое пристальное внимание на п. 3.1 регламента работы сервиса (и, в особенности, на его последний абзац).
В противном случае ваши вопросы будут удаляться по причине тег-спама, а систематические нарушения приведут к блокировке учетной записи.

Answer 1

[nrgian]

Гуглить "jpeg python library"
Мне выдало первой же ссылкой что нужно.

Их несколько - выбрать что больше нравится.

Comments

[YukiLina]

Там есть ответы только как загрузить картинку из памяти компьютера и что-либо с ней делать. А мне нужно именно создать файл и туда загрузить рисунок из программы

[nrgian]

YukiLina, library - это готовый набор функций для решения какого-то класса задач.
это не "там только ответы".
это не ответы.
это готовый набор функций, которым вы можете воспользоваться.

проверил.
повторил гуглеж по фразе "png python library"

первая же ссылки ведет в
https://pypi.org/project/pypng/

Переходим в раздел "Документация".
Далее переходим в раздел "Запись". То есть сюда https://pypng.readthedocs.io/en/latest/ex.html#writing

проверил - да, это то что вам и надо

[YukiLina]

nrgian, Спасибо)

Answer 2

[Владимир Куц]

from matplotlib.pyplot import imshow
%matplotlib inline
import cv2
import numpy as np

RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)


# создать изображение 100x100
image = np.zeros((100, 100, 3), np.uint8)

# залить его зеленым цветом
color = GREEN
out_color = tuple(reversed(color))
image[:] = out_color

# вывести на экран
imshow(image)

# сохранить в файл image.jpg
cv2.imwrite('image.jpg', image)

Если надо прям в памяти хранить файл в формате jpg- то дополняете таким кодом:

is_success, buffer = cv2.imencode(".jpg", image)
io_buf = io.BytesIO(buffer)

# decode
decode_img = cv2.imdecode(np.frombuffer(io_buf.getbuffer(), np.uint8), -1)

Comments

[YukiLina]

Я все перепробовала и все равно ничего не выходит. У меня не получается правильно изменить код. Вот например похожая программа :

from tkinter import *
 
 
canvas_width = 700
canvas_height = 500
brush_size = 3
color = 'black'
 
 
def paint(event):
    global brush_size
    global color
    x1 = event.x - brush_size
    x2 = event.x + brush_size
    y1 = event.y - brush_size
    y2 = event.y + brush_size
    w.create_oval(x1, y1, x2, y2,
                  fill = color, outline = color)
 
def brish_size_change(new_size):
    global brush_size
    brush_size = new_size
 
def color_change (new_color):
    global color
    color = new_color
 
 
root = Tk()
root.title('Paint')
 
w = Canvas(root,
           width = canvas_width,
           height = canvas_height,
           bg='white',
           cursor = "pencil")
 
 
 
w.bind('<B1-Motion>', paint)
 
 
black_btn = Button(text = "Чёрный", width = 10,
                   command=lambda: color_change("black"))
 
red_btn = Button(text = 'Красный', width = 10,
                 command = lambda:color_change("red"))
 
blue_btn = Button(text = 'Синий', width = 10,
                 command = lambda:color_change("blue"))
 
white_btn = Button(text = "Ластик", width = 10,
                   command=lambda: color_change("white"))
 
clear_btn = Button(text = "Удалить всё", width = 10,
                   command=lambda: w.delete("all"))
 
 
 
w.grid(row = 2, column = 0,
       columnspan = 7, padx = 5,
       pady = 5, sticky = E+W+S+N)
w.columnconfigure(6, weight = 1)
w.rowconfigure(2, weight = 1)
 
 
 
black_btn.grid(row = 0, column = 1)
red_btn.grid(row = 0, column = 2)
blue_btn.grid(row = 0, column = 3)
white_btn.grid(row = 0, column = 4)
clear_btn.grid(row = 0, column = 6, sticky = W)
 
root.mainloop()

Напишите кто может пожалуйста код, который сохраняет изображение нарисованное в canvas в память компьютера

[Александр]

Вот код с гитхаб немного доработанный

from tkinter import *
from tkinter.colorchooser import askcolor
from PIL import ImageGrab


class Paint(object):

    DEFAULT_PEN_SIZE = 5.0
    DEFAULT_COLOR = 'black'

    def __init__(self):
        self.root = Tk()

        self.pen_button = Button(self.root, text='pen', command=self.use_pen)
        self.pen_button.grid(row=0, column=0)

        self.brush_button = Button(self.root, text='brush', command=self.use_brush)
        self.brush_button.grid(row=0, column=1)

        self.color_button = Button(self.root, text='color', command=self.choose_color)
        self.color_button.grid(row=0, column=2)

        self.eraser_button = Button(self.root, text='eraser', command=self.use_eraser)
        self.eraser_button.grid(row=0, column=3)

        self.choose_size_button = Scale(self.root, from_=1, to=10, orient=HORIZONTAL)
        self.choose_size_button.grid(row=0, column=5)

        self.save_button = Button(self.root, text='save', command=self.saveImage)
        self.save_button.grid(row=0, column=4)

        self.c = Canvas(self.root, bg='white', width=600, height=600)
        self.c.grid(row=1, columnspan=5)

        self.setup()
        self.root.mainloop()

    def setup(self):
        self.old_x = None
        self.old_y = None
        self.line_width = self.choose_size_button.get()
        self.color = self.DEFAULT_COLOR
        self.eraser_on = False
        self.active_button = self.pen_button
        self.c.bind('<B1-Motion>', self.paint)
        self.c.bind('<ButtonRelease-1>', self.reset)

    def use_pen(self):
        self.activate_button(self.pen_button)

    def use_brush(self):
        self.activate_button(self.brush_button)

    def choose_color(self):
        self.eraser_on = False
        self.color = askcolor(color=self.color)[1]

    def use_eraser(self):
        self.activate_button(self.eraser_button, eraser_mode=True)

    def activate_button(self, some_button, eraser_mode=False):
        self.active_button.config(relief=RAISED)
        some_button.config(relief=SUNKEN)
        self.active_button = some_button
        self.eraser_on = eraser_mode

    def paint(self, event):
        self.line_width = self.choose_size_button.get()
        paint_color = 'white' if self.eraser_on else self.color
        if self.old_x and self.old_y:
            self.c.create_line(self.old_x, self.old_y, event.x, event.y,
                               width=self.line_width, fill=paint_color,
                               capstyle=ROUND, smooth=TRUE, splinesteps=36)
        self.old_x = event.x
        self.old_y = event.y

    def reset(self, event):
        self.old_x, self.old_y = None, None

    def saveImage(self):
        widget = self.c
        x = self.root.winfo_rootx() + widget.winfo_x()
        y = self.root.winfo_rooty()+ widget.winfo_y()
        x1 = x + widget.winfo_width()
        y1 = y + widget.winfo_height()
        ImageGrab.grab().crop((x, y, x1, y1)).save("paint.png")


if __name__ == '__main__':
    Paint()

[Александр]

Что бы именно в память то через numpy и PIL