Tripcode collision preventer

boards/abstracts/settingsmanager.py

@@ -1,154 +1,174 b''
 import uuid
-from django.shortcuts import get_object_or_404
+
+import boards
+from boards.abstracts.tripcode import Tripcode
+
 from boards.models import Tag
 
+MAX_TRIPCODE_COLLISIONS = 50
+
 __author__ = 'neko259'
 
 SESSION_SETTING = 'setting'
 
 # Remove this, it is not used any more cause there is a user's permission
 PERMISSION_MODERATE = 'moderator'
 
 SETTING_THEME = 'theme'
 SETTING_FAVORITE_TAGS = 'favorite_tags'
 SETTING_HIDDEN_TAGS = 'hidden_tags'
 SETTING_PERMISSIONS = 'permissions'
 SETTING_USERNAME = 'username'
 SETTING_LAST_NOTIFICATION_ID = 'last_notification'
 SETTING_IMAGE_VIEWER = 'image_viewer'
 SETTING_TRIPCODE = 'tripcode'
 
 DEFAULT_THEME = 'md'
 
 
 class SettingsManager:
     """
     Base settings manager class. get_setting and set_setting methods should
     be overriden.
     """
     def __init__(self):
         pass
 
     def get_theme(self) -> str:
         theme = self.get_setting(SETTING_THEME)
         if not theme:
             theme = DEFAULT_THEME
             self.set_setting(SETTING_THEME, theme)
 
         return theme
 
     def set_theme(self, theme):
         self.set_setting(SETTING_THEME, theme)
 
     def has_permission(self, permission):
         permissions = self.get_setting(SETTING_PERMISSIONS)
         if permissions:
             return permission in permissions
         else:
             return False
 
     def get_setting(self, setting, default=None):
         pass
 
     def set_setting(self, setting, value):
         pass
 
     def add_permission(self, permission):
         permissions = self.get_setting(SETTING_PERMISSIONS)
         if not permissions:
             permissions = [permission]
         else:
             permissions.append(permission)
         self.set_setting(SETTING_PERMISSIONS, permissions)
 
     def del_permission(self, permission):
         permissions = self.get_setting(SETTING_PERMISSIONS)
         if not permissions:
             permissions = []
         else:
             permissions.remove(permission)
         self.set_setting(SETTING_PERMISSIONS, permissions)
 
     def get_fav_tags(self) -> list:
         tag_names = self.get_setting(SETTING_FAVORITE_TAGS)
         tags = []
         if tag_names:
             tags = list(Tag.objects.filter(name__in=tag_names))
         return tags
 
     def add_fav_tag(self, tag):
         tags = self.get_setting(SETTING_FAVORITE_TAGS)
         if not tags:
             tags = [tag.name]
         else:
             if not tag.name in tags:
                 tags.append(tag.name)
 
         tags.sort()
         self.set_setting(SETTING_FAVORITE_TAGS, tags)
 
     def del_fav_tag(self, tag):
         tags = self.get_setting(SETTING_FAVORITE_TAGS)
         if tag.name in tags:
             tags.remove(tag.name)
             self.set_setting(SETTING_FAVORITE_TAGS, tags)
 
     def get_hidden_tags(self) -> list:
         tag_names = self.get_setting(SETTING_HIDDEN_TAGS)
         tags = []
         if tag_names:
             tags = list(Tag.objects.filter(name__in=tag_names))
 
         return tags
 
     def add_hidden_tag(self, tag):
         tags = self.get_setting(SETTING_HIDDEN_TAGS)
         if not tags:
             tags = [tag.name]
         else:
             if not tag.name in tags:
                 tags.append(tag.name)
 
         tags.sort()
         self.set_setting(SETTING_HIDDEN_TAGS, tags)
 
     def del_hidden_tag(self, tag):
         tags = self.get_setting(SETTING_HIDDEN_TAGS)
         if tag.name in tags:
             tags.remove(tag.name)
             self.set_setting(SETTING_HIDDEN_TAGS, tags)
 
     def get_tripcode(self):
-        return self.get_setting(SETTING_TRIPCODE, str(uuid.uuid4()))
+        tripcode = self.get_setting(SETTING_TRIPCODE)
+        if tripcode is None:
+            self.reset_tripcode()
+            tripcode = self.get_setting(SETTING_TRIPCODE)
+        return tripcode
 
     def reset_tripcode(self):
-        self.set_setting(SETTING_TRIPCODE, str(uuid.uuid4()))
+        tripcode = Tripcode(str(uuid.uuid4()))
+
+        # If we cannot find a collision-free tripcode, then let the collision
+        # be destiny
+        collision_counter = 0
+        while boards.models.Post.objects.filter(
+                tripcode__startswith=tripcode.get_short_text()).exists()\
+                and collision_counter < MAX_TRIPCODE_COLLISIONS:
+            tripcode = Tripcode(str(uuid.uuid4()))
+            collision_counter += 1
+            print('Tripcode collision detected')  # FIXME Use proper logging
+        self.set_setting(SETTING_TRIPCODE, tripcode.get_full_text())
 
 
 class SessionSettingsManager(SettingsManager):
     """
     Session-based settings manager. All settings are saved to the user's
     session.
     """
     def __init__(self, session):
         SettingsManager.__init__(self)
         self.session = session
 
     def get_setting(self, setting, default=None):
         if setting in self.session:
             return self.session[setting]
         else:
             self.set_setting(setting, default)
             return default
 
     def set_setting(self, setting, value):
         self.session[setting] = value
 
 
 def get_settings_manager(request) -> SettingsManager:
     """
     Get settings manager based on the request object. Currently only
     session-based manager is supported. In the future, cookie-based or
     database-based managers could be implemented.
     """
     return SessionSettingsManager(request.session)

boards/abstracts/tripcode.py

@@ -1,27 +1,30 b''
 class Tripcode:
     def __init__(self, code_str):
         self.tripcode = code_str
 
     def get_color(self):
         return self.tripcode[:6]
 
     def get_background(self):
         code = self.get_color()
         result = ''
 
         for i in range(0, len(code), 2):
             p = code[i:i+2]
             background = hex(255 - int(p, 16))[2:]
             if len(background) < 2:
                 background = '0' + background
             result += background
 
         return result
 
     def get_short_text(self):
         return self.tripcode[:8]
 
+    def get_full_text(self):
+        return self.tripcode
+
     def get_view(self):
-        return '<span style="color: #{}; background: #{}">{}</span>'\
-            .format(self.get_color(), self.get_background(),
-                    self.get_short_text())
  No newline at end of file
+        return '<span title="{}" style="color: #{}; background: #{}">{}</span>'\
+            .format(self.get_full_text(), self.get_color(), self.get_background(),
+                    self.get_short_text())