Prevent generation of very long names for lock file (#97)

HISTORY.rst

@@ -1,11 +1,22 @@
 History
 =======
 
+3.0.0
+-----
+2019-05-13
+
+Fixed an issue where large/long arguments could cause ``OSError Filename too long`` with the file backend (see #96).
+Keys generated for file backend, are now hashed and limited to 50 characters in length.
+*Due to this, it is not backwards compatible with existing keys from the file backend, so any pending locks from previous version will be ignored.*
+The Redis backend is unchanged, and thus fully compatible.
+
+Credit for fix to @xuhcc.
+
 2.1.2
 -----
 2019-05-13
 
-- Add support for 'rediss'. Thanks @gustavoalmeida
+- Add support for ``rediss``. Thanks @gustavoalmeida
 
 2.1.1
 -----

celery_once/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = 'Cameron Maske'
 __email__ = '[email protected]'
-__version__ = '2.1.2'
+__version__ = '3.0.0'
 
 
 from .tasks import QueueOnce, AlreadyQueued

celery_once/backends/file.py

@@ -1,14 +1,27 @@
 """
 Definition of the file locking backend.
 """
+import hashlib
 import errno
 import os
 import tempfile
 import time
 
+import six
+
 from celery_once.tasks import AlreadyQueued
 
 
+def key_to_lock_name(key):
+    """
+    Combine part of a key with its hash to prevent very long filenames
+    """
+    MAX_LENGTH = 50
+    key_hash = hashlib.md5(six.b(key)).hexdigest()
+    lock_name = key[:MAX_LENGTH - len(key_hash) - 1] + '_' + key_hash
+    return lock_name
+
+
 class File(object):
     """
     File locking backend.
@@ -27,7 +40,8 @@ def __init__(self, settings):
                 raise
 
     def _get_lock_path(self, key):
-        return os.path.join(self.location, key)
+        lock_name = key_to_lock_name(key)
+        return os.path.join(self.location, lock_name)
 
     def raise_or_lock(self, key, timeout):
         """

tests/integration/backends/test_file.py

@@ -23,7 +23,7 @@ def example(a=1):
 
 @pytest.fixture()
 def lock_path():
-    path = '/tmp/celery_once/qo_example_a-1'
+    path = '/tmp/celery_once/qo_example_a-1_b7f89d8561e5788a3e7687c6ede93bcd'
     yield path
     os.remove(path) # Remove file after test function runs.
 

tests/unit/backends/test_file.py

@@ -5,10 +5,17 @@
 
 import pytest
 
-from celery_once.backends.file import File
+from celery_once.backends.file import key_to_lock_name, File
 from celery_once.tasks import AlreadyQueued
 
 
+def test_key_to_lock_name():
+    assert key_to_lock_name('qo_test') == \
+        'qo_test_999f583e69db6a0c04b86beeebb2b631'
+    assert key_to_lock_name('qo_looooooong_task_name') == \
+        'qo_looooooong_tas_6626e5965e549303044d5a7f4fdc3c6b'
+
+
 def test_file_init(mocker):
     makedirs_mock = mocker.patch('celery_once.backends.file.os.makedirs')
     location = '/home/test'
@@ -55,7 +62,8 @@ def test_file_create_lock(backend, mocker):
     mtime_mock = mocker.patch('celery_once.backends.file.os.path.getmtime')
     utime_mock = mocker.patch('celery_once.backends.file.os.utime')
     close_mock = mocker.patch('celery_once.backends.file.os.close')
-    expected_lock_path = os.path.join(TEST_LOCATION, key)
+    expected_lock_path = os.path.join(TEST_LOCATION,
+                                      key_to_lock_name(key))
     ret = backend.raise_or_lock(key, timeout)
 
     assert open_mock.call_count == 1
@@ -103,7 +111,8 @@ def test_file_lock_timeout(backend, mocker):
         return_value=1550156000.0)
     utime_mock = mocker.patch('celery_once.backends.file.os.utime')
     close_mock = mocker.patch('celery_once.backends.file.os.close')
-    expected_lock_path = os.path.join(TEST_LOCATION, key)
+    expected_lock_path = os.path.join(TEST_LOCATION,
+                                      key_to_lock_name(key))
     ret = backend.raise_or_lock(key, timeout)
 
     assert open_mock.call_count == 1
@@ -115,7 +124,8 @@ def test_file_lock_timeout(backend, mocker):
 def test_file_clear_lock(backend, mocker):
     key = 'test.task.key'
     remove_mock = mocker.patch('celery_once.backends.file.os.remove')
-    expected_lock_path = os.path.join(TEST_LOCATION, key)
+    expected_lock_path = os.path.join(TEST_LOCATION,
+                                      key_to_lock_name(key))
     ret = backend.clear_lock(key)
 
     assert remove_mock.call_count == 1