support regex patterns in projectmap config section

tests/samples/configs/project_map.cfg

@@ -2,3 +2,5 @@
 debug = false
 api_key = 1234
 [projectmap]
+samples/projects/proj.{3}_map/ = proj-map
+samples/projects/project_map(\d+)/ = proj-map{0}

tests/test_project.py

@@ -160,18 +160,27 @@ class LanguagesTestCase(utils.TestCase):
         response.status_code = 0
         self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
 
-        with tempfile.NamedTemporaryFile() as fh:
         now = u(int(time.time()))
         entity = 'tests/samples/projects/project_map/emptyfile.txt'
-
+        config = 'tests/samples/configs/project_map.cfg'
-            fh.write(open('tests/samples/configs/project_map.cfg').read().encode('utf-8'))
-            fh.write('{0} = proj-map'.format(os.path.realpath(os.path.dirname(os.path.dirname(entity)))).encode('utf-8'))
-            fh.flush()
-
-            config = fh.name
 
         args = ['--file', entity, '--config', config, '--time', now]
 
         execute(args)
 
         self.assertEquals('proj-map', self.patched['wakatime.offlinequeue.Queue.push'].call_args[0][0]['project'])
+
+    def test_project_map_group_usage(self):
+        response = Response()
+        response.status_code = 0
+        self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
+
+        now = u(int(time.time()))
+        entity = 'tests/samples/projects/project_map42/emptyfile.txt'
+        config = 'tests/samples/configs/project_map.cfg'
+
+        args = ['--file', entity, '--config', config, '--time', now]
+
+        execute(args)
+
+        self.assertEquals('proj-map42', self.patched['wakatime.offlinequeue.Queue.push'].call_args[0][0]['project'])

wakatime/projects/projectmap.py

@@ -3,18 +3,18 @@
     wakatime.projects.projectmap
     ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
-    Use the ~/.wakatime.cfg file to set custom project names by
+    Use the ~/.wakatime.cfg file to set custom project names by matching files
-    recursively matching folder paths.
+    with regex patterns. Project maps go under the [projectmap] config section.
-    Project maps go under the [projectmap] config section.
 
     For example:
 
         [projectmap]
         /home/user/projects/foo = new project name
-        /home/user/projects/bar = project2
+        /home/user/projects/bar(\d+)/ = project{0}
 
     Will result in file `/home/user/projects/foo/src/main.c` to have
-    project name `new project name`.
+    project name `new project name` and file `/home/user/projects/bar42/main.c`
+    to have project name `project42`.
 
     :copyright: (c) 2013 Alan Hamlett.
     :license: BSD, see LICENSE for more details.
@@ -22,6 +22,7 @@
 
 import logging
 import os
+import re
 
 from .base import BaseProject
 from ..compat import u
@@ -42,20 +43,23 @@ class ProjectMap(BaseProject):
 
     def _find_project(self, path):
         path = os.path.realpath(path)
-        if os.path.isfile(path):
-            path = os.path.split(path)[0]
 
-        if self._configs.get(path.lower()):
+        try:
-            return self._configs.get(path.lower())
+            for pattern, new_proj_name in self._configs.items():
-        if self._configs.get('%s/' % path.lower()):  # pragma: nocover
+                try:
-            return self._configs.get('%s/' % path.lower())
+                    compiled = re.compile(pattern, re.IGNORECASE)
-        if self._configs.get('%s\\' % path.lower()):  # pragma: nocover
+                    match = compiled.search(path)
-            return self._configs.get('%s\\' % path.lower())
+                    if match:
+                        return new_proj_name.format(*match.groups())
+                except re.error as ex:
+                    log.warning(u('Regex error ({msg}) for projectmap pattern: {pattern}').format(
+                        msg=u(ex),
+                        pattern=u(pattern),
+                    ))
+        except TypeError:  # pragma: nocover
+            pass
 
-        split_path = os.path.split(path)
+        return None
-        if split_path[1] == '':
-            return None  # pragma: nocover
-        return self._find_project(split_path[0])
 
     def branch(self):
         return None
@@ -63,4 +67,4 @@ class ProjectMap(BaseProject):
     def name(self):
         if self.project:
             return u(self.project)
-        return None  # pragma: nocover
+        return None