adding logic for pre

Author: choldgraf

docs/index.md

@@ -22,7 +22,7 @@ The easiest way to install this package is to do so directly from GitHub with `p
 pip install git+https://github.com/choldgraf/github-activity
 ```
 
-## Usage
+## Generate a markdown changelog
 
 The easiest way to use `github-activity` to generate activity markdown is to use
 the command-line interface. It takes the following form:
@@ -50,6 +50,11 @@ github-activity jupyter/notebook -s 6.0.0 -u 6.0.1 -o sample_notebook_activity.m
 
 You can find the [resulting markdown here](sample_notebook_activity).
 
+### Splitting PRs by tags and prefixes
+
+Often you wish to split your PRs into multiple categories. `github-activity`
+follows the [keepachangelog taxonomy of changes](https://keepachangelog.com/en/1.0.0/).
+
 ### Using a GitHub API token
 
 `github-activity` uses the GitHub API to pull information about a repository's activity.

github_activity/github_activity.py

@@ -16,12 +16,12 @@
 TAGS_METADATA_BASE = {
     "new": {
         "tags": ["feature", "new"],
-        "pre": ["NEW"],
+        "pre": ["NEW", "FEAT", "FEATURE"],
         "description": "New features added",
     },
     "enhancement": {
-        "tags": ["enhancement", "feature", "enhancements"],
-        "pre": ["NEW", "ENH"],
+        "tags": ["enhancement", "enhancements"],
+        "pre": ["NEW", "ENH", "ENHANCEMENT", "IMPROVE"],
         "description": "Enhancements made",
     },
     "bug": {
@@ -31,7 +31,7 @@
     },
     "maintenance": {
         "tags": ["maintenance", "maint"],
-        "pre": ["MAINT"],
+        "pre": ["MAINT", "MNT"],
         "description": "Maintenance and upkeep improvements",
     },
     "documentation": {
@@ -41,12 +41,12 @@
     },
     "api_change": {
         "tags": ["api-change", "apichange"],
-        "tags": ["BREAK", "BREAKING", "UPGRADE"],
+        "pre": ["BREAK", "BREAKING", "BRK", "UPGRADE"],
         "description": "API and Breaking Changes",
     },
     "deprecate": {
         "tags": ["deprecation", "deprecate"],
-        "tags": ["DEPRECATE", "DEPRECATION"],
+        "pre": ["DEPRECATE", "DEPRECATION", "DEP"],
         "description": "Deprecated features",
     },
 }
@@ -294,9 +294,16 @@ def generate_activity_md(
 
     # Separate out items by their tag types
     for kind, kindmeta in tags_metadata.items():
+        # First find the PRs based on tag
         mask = closed_prs["labels"].map(
             lambda a: any(ii in jj for ii in kindmeta["tags"] for jj in a)
         )
+        # Now find PRs based on prefix
+        mask_pre = closed_prs["title"].map(
+            lambda title: any(f"{ipre}:" in title for ipre in kindmeta["pre"])
+        )
+        mask = mask | mask_pre
+
         kindmeta["data"] = closed_prs.loc[mask]
         kindmeta["mask"] = mask