From b7fbc3f98417fb937098f5e00dc269ecb47349b5 Mon Sep 17 00:00:00 2001
From: Bin Yang <bin.yang@algotune.com>
Date: Mon, 12 Apr 2021 14:27:25 -0400
Subject: [PATCH] added search terms for data/portfolio construction/factor

---
 .github/workflows/repo_status.yml |  2 +-
 .github/workflows/wiki_gen.yml    | 10 +++++-----
 git_status.py                     | 30 ++++++++++++++++++++++++++++++
 3 files changed, 36 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/repo_status.yml b/.github/workflows/repo_status.yml
index ab10d78..61f8cb0 100644
--- a/.github/workflows/repo_status.yml
+++ b/.github/workflows/repo_status.yml
@@ -21,7 +21,7 @@ jobs:
         uses: actions/setup-python@v2
         with:
           python-version: 3.7
-
+d
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
diff --git a/.github/workflows/wiki_gen.yml b/.github/workflows/wiki_gen.yml
index cb734e2..8425de3 100644
--- a/.github/workflows/wiki_gen.yml
+++ b/.github/workflows/wiki_gen.yml
@@ -25,11 +25,11 @@ jobs:
           python -m pip install --upgrade pip
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
-#      - name: execute py script # run the run.py to get the latest data
-#        run: |
-#          python git_status.py
-#        env:
-#          GIT_TOKEN: ${{ secrets.GIT_TOKEN }}
+      - name: execute py script # run the run.py to get the latest data
+        run: |
+          python git_status.py
+        env:
+          GIT_TOKEN: ${{ secrets.GIT_TOKEN }}
 
       - name: execute py script # run the wiki_gen
         run: |
diff --git a/git_status.py b/git_status.py
index 28e02d8..49e3ff7 100644
--- a/git_status.py
+++ b/git_status.py
@@ -154,6 +154,36 @@ def search_new_repo_by_category(category: str,
                                                  category,
                                                  min_stars_number=min_stars_number
                                                  )
+    elif category == 'Data Processing Techniques and Transformations':
+        combined_df = search_repo_multiple_terms(['data transformation trading',
+                                                  'data transformation finance',
+                                                  'data transformation time series'],
+                                                 category,
+                                                 min_stars_number=min_stars_number
+                                                 )
+    elif category == 'Portfolio Selection and Optimisation':
+        combined_df = search_repo_multiple_terms(['portfolio optimization machine learning finance',
+                                                  'portfolio optimization machine learning trading',
+                                                  'portfolio construction machine learning finance'
+                                                  'portfolio construction machine learning trading',
+                                                  'portfolio optimization finance',
+                                                  'portfolio optimization trading'],
+                                                 category,
+                                                 min_stars_number=min_stars_number
+                                                 )
+    elif category == 'Factor and Risk Analysis':
+        combined_df = search_repo_multiple_terms(['risk factor finance',
+                                                  'risk factor trading',
+                                                  'risk premia factor finance'
+                                                  'risk premia factor trading',
+                                                  'style factor finance',
+                                                  'style factor trading',
+                                                  'macro factor finance',
+                                                  'macro factor trading'
+                                                  ],
+                                                 category,
+                                                 min_stars_number=min_stars_number
+                                                 )
 
     # only find ones that need to be inserted
     if combined_df is not None and existing_repo_df is not None: