Add an option named g:leetcode_problemset to select the problemset.

Author: tenfyzhong

README.md

@@ -106,6 +106,12 @@ Hide the companies section.
 
 Default value is `0`
 
+### `g:leetcode_problemset`
+
+Set the problemset to get from leetcode. 
+
+Default value is `all`
+
 ## Updates
 
 - 2019/12/20: Fix the login issue caused by reCAPTCHA.

autoload/leetcode.vim

@@ -367,15 +367,16 @@ function! s:ListProblemsOfCompany(company_slug, refresh) abort
 endfunction
 
 function! leetcode#ListProblems(refresh) abort
-    let buf_name = 'leetcode:///problems/all'
+    let buf_name = 'leetcode:///problems/' . g:leetcode_problemset
     if s:CheckSignIn() == v:false
         return
     endif
+    let problemset = printf('leetcode.get_problems(["%s"])', g:leetcode_problemset)
     if buflisted(buf_name)
         execute bufnr(buf_name) . 'buffer'
         let saved_view = winsaveview()
         if a:refresh ==# 'redownload'
-            let expr = 'leetcode.get_problems(["all"])'
+            let expr = problemset
             let b:leetcode_downloaded_problems = py3eval(expr)
         elseif a:refresh ==# 'norefresh'
             return
@@ -387,7 +388,7 @@ function! leetcode#ListProblems(refresh) abort
         execute 'rightbelow new ' . buf_name
         call s:SetupProblemListBuffer()
         let b:leetcode_buffer_type = 'all'
-        let expr = 'leetcode.get_problems(["all"])'
+        let expr = problemset
         let b:leetcode_downloaded_problems = py3eval(expr)
         let b:leetcode_difficulty = 'All'
         let b:leetcode_state = 'All'

doc/leetcode.txt

@@ -97,4 +97,10 @@ g:leetcode_hide_companies
 
                     Default: 0
 
+                                                      *g:leetcode_problemset*
+g:leetcode_problemset
+                    Set the problemset to get from leetcode.
+
+                    Default: 'all'
+
  vim:tw=78:ts=8:et:ft=help:norl:

plugin/leetcode.vim

@@ -32,6 +32,10 @@ if !exists('g:leetcode_hide_companies')
     let g:leetcode_hide_companies = 0
 endif
 
+if !exists('g:leetcode_problemset')
+    let g:leetcode_problemset = 'all'
+endif
+
 command! -nargs=0 LeetCodeList call leetcode#ListProblems('redraw')
 command! -nargs=0 LeetCodeReset call leetcode#ResetSolution(0)
 command! -nargs=0 LeetCodeTest call leetcode#TestSolution()