funcs.py

from PIL import Image #Necesario para redimensionar la imagen
from os import path
from resizeimage import resizeimage #Necesario para redimensionar la imagen
from variables import Subreddits_list, PostsToAvoid

def process_title(titulo): #Procesa los titulos de los posts en reddit.
    if(" [OC]" in titulo):
        return titulo.replace(" [OC]", "")
    elif("[OC] " in titulo):
        return titulo.replace("[OC] ", "")
    elif("[OC]" in titulo):
        return titulo.replace("[OC]", "")
    else:
        return titulo

def Redimensionar(img_path): #Redimensiona la imagen si se excede de los limites de resolucion de twitter 8192x8192
    RES_LIMIT= 1920
    with Image.open(img_path) as image:
        width, height = image.size
        if(width > RES_LIMIT or height > RES_LIMIT):
            print("Se redimensiona la imagen")
            new_size = (RES_LIMIT, RES_LIMIT)
            try:
                image.thumbnail(new_size, Image.ANTIALIAS)
                image.save(img_path, optimize= True, quality= 85)

                #Tratar de reducir el tamaño del archivo. Si despues de tres veces sigue excediendo 3072kb, devuelve Falso
                if((path.getsize(img_path) / 1024) > 3072):
                    for _ in range(3):
                        image.save(img_path, optimize= True, quality= 95)
                        if((path.getsize(img_path) / 1024) < 3072):
                            break
                    else: #Si sigue excediendo 3072kb
                        print("La imagen sigue teniendo un tamaño de archivo muy grande: " + str(path.getsize(img_path) / 1024) + " kb.")
                        return False

            except Exception as e:
                print("Ha ocurrido una excepción al redimensionar la imagen: " + str(e))
                return False
        else:
            print("No es necesario redimensionar la imagen")
    return True

def isOriginal(titulo, TWEETS): #Comprueba si el post elegido para publicarse ya fue publicado en el timeline del bot
    for status in TWEETS:
        if(process_title(titulo) in status):
            break
    else: #Si no se ejecuta el break
        return True
    return False

class Scheduling():
    """ 
    Clase que organiza las funciones dedicadas a 
    planificar en que orden se van a sacar las fotos de
    los subreddits
    """
    def __init__(self, PATH):
        self.path = PATH

    def __OpenFile(self):
        try:
            file = open(self.path, "r+") #Abrir para lectura y escritura
        except FileNotFoundError:
            file = open(self.path, "w") #Lo crea
            file.close()
            file = open(self.path, "r+") #Abrir para lectura y escritura
        return file

    def __get_LastSub(self):
        f = self.__OpenFile()
        LastSub = f.read()
        f.close()
        return LastSub

    def get_NextSub(self):
        contents = self.__get_LastSub()
        if(contents == ""):
            contents = Subreddits_list[0].lower().capitalize()

        NextSubIndex= Subreddits_list.index(contents) + 1
        try:
            NextSub = Subreddits_list[NextSubIndex]
        except:
            NextSub = Subreddits_list[0]
        return NextSub

    def set_NewSub(self, newSub):
        if(newSub.lower().capitalize() not in Subreddits_list):
            raise Exception #El subreddit especificado no forma parte de la lista de subreddits.

        f = self.__OpenFile()
        f.seek(0) #Cambia la posicion dentro del archivo al principio
        f.truncate() #Borra todo lo que exceda la posicion dentro del archivo
        f.write(newSub.lower().capitalize())
        f.close()
        return None

def CheckInPosts(POST):
    #Verificar si el post que se quiere publicar
    #presenta una excepcion por tener un texto largo.
    if(POST in PostsToAvoid):
        return False
    else:
        return True