Skip to content
Snippets Groups Projects
Commit 617f8682 authored by dnns01's avatar dnns01
Browse files

Fix link protection when nick contains capital letters

parent 0b8bb17d
Branches
No related tags found
No related merge requests found
......@@ -9,6 +9,11 @@ class TextCommand(models.Model):
text = models.TextField(max_length=500)
active = models.BooleanField(default=True)
def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
if self.command[0] == "!":
self.command = self.command[1:]
super().save(force_insert, force_update, using, update_fields)
class Klassenbuch(models.Model):
name = models.CharField(max_length=50)
......@@ -23,6 +28,10 @@ class Timer(models.Model):
class LinkPermit(models.Model):
nick = models.CharField(max_length=25)
def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
self.nick = self.nick.lower()
super().save(force_insert, force_update, using, update_fields)
class LinkWhitelist(models.Model):
url = models.URLField()
......
......@@ -2,9 +2,10 @@ import re
import sqlite3
from datetime import datetime, timedelta
import config
from twitchio.ext import commands
import config
@commands.core.cog(name="LinkProtection")
class LinkProtection:
......@@ -104,6 +105,8 @@ class LinkProtection:
if user[0] == "@":
user = user[1:]
user = user.lower()
if ctx.author.is_mod:
if permanent:
self.give_user_permanent_permit(user)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment