modules.kek

Complexity

Total Complexity

21

Size/Duplication

Total Lines	119
Duplicated Lines	0 %

Importance

Changes

0

6 Functions

Rating	Name	Duplication	Size	Complexity
A	module_init()	0	8	2
B	kekify()	0	20	5
A	get_values()	0	13	2
A	multikek()	0	16	1
B	kek()	0	31	7
A	kekify_gifs()	0	10	4

#!/usr/bin/python
# -*- coding: utf-8 -*-
from modules.utils import caption_filter, get_image, send_image
from modules.logging import logging_decorator
from telegram.ext import CommandHandler, MessageHandler
from telegram import ChatAction
from wand.image import Image
from datetime import datetime
import os


def module_init(gd):
    global path, extensions
    extensions = gd.config["extensions"]
    commands = gd.config["commands"]
    path = gd.config["path"]
    for command in commands:
        gd.dp.add_handler(MessageHandler(caption_filter("/"+command), kek))
        gd.dp.add_handler(CommandHandler(command, kek))


@logging_decorator("kek")
def kek(bot, update):
    filename = datetime.now().strftime("%d%m%y-%H%M%S%f")
    if update.message.reply_to_message is not None:
        kek_param = "".join(update.message.text[5:7])
    elif update.message.caption is not None:
        kek_param = "".join(update.message.caption[5:7])
    else:
        update.message.reply_text("You need an image for that")
        return
    try:
        extension = get_image(bot, update, path, filename)

The variable path does not seem to be defined.

    except:
        update.message.reply_text("Can't get the image")
        return
    if extension not in extensions:

The variable extensions does not seem to be defined.

        update.message.reply_text("Unsupported file")
        return False
    update.message.chat.send_action(ChatAction.UPLOAD_PHOTO)
    
    if extension in [".mp4", ".gif"]:
        if kek_param == "-m":
            update.message.reply_text("Multikek unsupported for animations")
            return
        result = kekify_gifs(kek_param, filename, extension)
    else:
        _, _, result = kekify(kek_param, filename, extension, None)
    result.save(filename=path+filename+extension)
    result.close()
    send_image(update, path, filename, extension)
    os.remove(path+filename+extension)


def kekify(kek_param, filename, extension, file):
    if kek_param == "-m":
        result = multikek(filename, extension)
        return None, None, result
    if file == None:
        source = Image(filename=path+filename+extension)

The variable path does not seem to be defined.

    else:
        source = file
    w = source.width; h = source.height
    c, p1, p2, f = get_values(kek_param, w, h)
    with source.clone() as part1:
        part1.crop(c[0], c[1], c[2], c[3])
        with part1.clone() as part2:
            getattr(part2, f)()
            new_canvas = Image()
            new_canvas.blank(w, h)
            new_canvas.composite(part1, p1[0], p1[1])
            new_canvas.composite(part2, p2[0], p2[1])
    source.close()
    return w, h, new_canvas
    
def kekify_gifs(kek_param, filename, extension):
    with Image(filename=path+filename+extension) as source:

The variable path does not seem to be defined.

        w = source.width; h = source.height
        new = Image()
        for i in range(len(source.sequence)):
            with source.sequence[i] as frame: 
                img = Image(image=frame)
                _, _, result = kekify(kek_param, filename, extension, img)
            new.sequence.append(result)
    return new


def multikek(filename, extension):
    w, h, canvasL = kekify("-l", filename, extension, None)
    w, h, canvasR = kekify("-r", filename, extension, None)
    w, h, canvasT = kekify("-t", filename, extension, None)
    w, h, canvasB = kekify("-b", filename, extension, None)
    big_canvas = Image()
    big_canvas.blank(w*2, h*2)
    big_canvas.composite(canvasL, 0, 0)
    big_canvas.composite(canvasR, w, 0)
    big_canvas.composite(canvasT, 0, h)
    big_canvas.composite(canvasB, w, h)
    canvasL.close()
    canvasR.close()
    canvasT.close()
    canvasB.close()
    return big_canvas


def get_values(kek_param, w, h):
    parameters = {
        "":   [[0, 0, w//2, h], [0, 0], [w//2, 0], "flop"],
        "-l": [[0, 0, w//2, h], [0, 0], [w//2, 0], "flop"],
        "-r": [[w//2, 0, w, h], [w//2, 0], [0, 0], "flop"],
        "-t": [[0, 0, w, h//2], [0, 0], [0, h//2], "flip"],
        "-b": [[0, h//2, w, h], [0, h//2], [0, 0], "flip"],
        }
    try:
        params = parameters[kek_param]
    except KeyError:
        params = parameters[""]
    return params