nxy/bot/drugs.py

# -*- coding: utf-8 -*-
import random

from docopt import Dict
from irc3.plugins.command import command
from irc3.utils import IrcString

from . import Plugin


class Drugs(Plugin):
    # All strains from https://www.leafly.com/explore
    STRAINS = {
        'sativa': [
            'Sour Diesel',
            'Green Crack',
            'Jack Herer',
            'Durban Poison',
            'Lemon Haze',
            'Strawberry Cough',
            'Super Silver Haze',
            'Alaskan Thunder Fuck',
            'Super Lemon Haze',
            'Amnesia Haze',
            'Maui Wowie',
            'Chocolope',
            'Harlequin',
        ],
        'indica': [
            'Granddaddy Purple',
            'Bubba Kush',
            'Northern Lights',
            'Blue Cheese',
            'Purple Kush',
            'Blueberry',
            'Grape Ape',
            'Blackberry Kush',
            'Master Kush',
            'God\'s Gift',
            'LA Confidential',
            'Death Star',
            'Purple Urkle',
            'Afghan Kush',
            'Skywalker',
            'White Rhino',
            'Hindu Kush',
        ],
        'hybrid': [
            'Blue Dream',
            'Girl Scout Cookies',
            'OG Kush',
            'White Widow',
            'Gorilla Glue #4',
            'Pineapple Express',
            'Trainwreck',
            'AK-47',
            'Headband',
            'Chemdawg',
            'Cheese',
            'Cherry Pie',
            'Skywalker OG',
            'Tahoe OG Kush',
            'Lemon Kush',
            'Platinum Girl Scout Cookies',
            'Golden Goat',
            'Agent Orange',
        ],
    }
    ACTIONS = {
        'joint': [
            ('passes', 'to'),
            ('rolls', 'for'),
            ('lights', 'for'),
        ],
        'bong': [
            ('passes', 'to'),
            ('preps', 'for'),
            ('lights', 'for'),
        ],
        'vape': [
            ('passes', 'to'),
        ],
        'blunt': [
            ('passes', 'to'),
            ('rolls', 'for'),
            ('lights', 'for'),
        ],
    }

    @command
    def dope(self, mask: IrcString, target: IrcString, args: Dict):
        """Smoke weed everyday

        %%dope [<nick>]
        """
        strain_types = list(self.STRAINS.keys())
        strain_type = random.choice(strain_types)
        consume_type = random.choice(list(self.ACTIONS.keys()))
        action = random.choice(self.ACTIONS[consume_type])

        self.bot.action(target, '{action0} a {consume} of the finest {type} "{strain}" {action1} {nick}'.format(
            action0=action[0],
            action1=action[1],
            consume=consume_type,
            strain=random.choice(self.STRAINS[strain_type]),
            nick=args.get('<nick>', mask.nick),
            type=strain_type,
        ))

    @command
    def meth(self, mask: IrcString, target: IrcString, args: Dict):
        """Snort some meth *_*

        %%meth [<nick>]
        """
        self.bot.action(target, r'legt {} eine dicke Line Meth \________'.format(args.get('<nick>', mask.nick)))