crystal-selfbot/utils.py at main · dropalways/crystal-selfbot · GitHub

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import json
import requests
import shutil
import ctypes
import requests
import os
def load_json(jsonfile):
    with open(jsonfile) as f:
        return json.load(f)

def set_console_title(title):
    try:
        ctypes.windll.kernel32.SetConsoleTitleW(title)
    except:
        pass

def clear_console():
    os.system('cls' if os.name=='nt' else 'clear')


def waifu_pics(profanity: str = "sfw", category: str=None):
    profanity_list = ["nsfw", "sfw"]
    nsfw_list = ["waifu", "neko", "trap", "blowjob"]
    sfw_list = [
    "waifu",
    "neko",
    "shinobu",
    "megumin",
    "bully",
    "cuddle",
    "cry",
    "hug",
    "awoo",
    "kiss",
    "lick",
    "pat",
    "smug",
    "bonk",
    "yeet",
    "blush",
    "smile",
    "wave",
    "highfive",
    "handhold",
    "nom",
    "bite",
    "glomp",
    "slap",
    "kill",
    "kick",
    "happy",
    "wink",
    "poke",
    "dance",
    "cringe"
    ]

    if profanity not in profanity_list:
        return f"Invalid profanity {profanity_list}"

    if category not in nsfw_list and profanity == "nsfw":
        return f"Invalid category {nsfw_list}"

    if category not in sfw_list and profanity == "sfw":
        return f"Invalid category {sfw_list}"

    request = requests.get(f"https://api.waifu.pics/{profanity}/{category}")
    match request.status_code:
        case 404:
            if profanity == "nsfw":
                return f"Error: Category '{category}' not found. {nsfw_list}"
            else:
                return f"Error: Category '{category}' not found. {sfw_list}"
        case 500:
            return f"Error: Category '{category}' internal server error."
        case 200:
            return request.json()["url"]
        case 429:
            return f"Error: Rate limit exceeded. Please wait for {request.headers['Retry-After']} seconds."
        case _:
            return "niggers"