Include neetcode shorts

Opt for UV instead of basic python

Author: johnsutor

.coveragerc

@@ -1,5 +1,5 @@
 [run]
-omit = tests/*,*/__init__.py
+omit = tests/*,*/__init__.py,scripts/*
 branch = True
 
 [report]

.github/workflows/check-style.yml

@@ -17,16 +17,19 @@ jobs:
     runs-on: '${{ matrix.os }}'
     steps:
     - uses: actions/checkout@v3
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v3
+    - name: Set up UV
+      uses: astral-sh/setup-uv@v3
       with:
-        python-version: ${{ matrix.python-version }}
+        version: "latest"
+    - name: Set up Python ${{ matrix.python-version }}
+      run: |
+        uv python install ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
-        python -m pip install -e ".[dev]"
+        uv sync --dev
     - name: Examine formatting with ruff 
       run: |
-        python -m ruff check ./leetcode_study_tool 
+        uv run ruff check ./leetcode_study_tool 
     - name: Check formatting 
       run: |
-        python -m ruff format ./leetcode_study_tool --check
+        uv run ruff format ./leetcode_study_tool --check

.github/workflows/publish.yml

@@ -15,16 +15,15 @@ jobs:
       id-token: write
     steps:
       - uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v4
+      - name: Set up UV
+        uses: astral-sh/setup-uv@v3
         with:
-          python-version: "3.x"
-      - name: Install dependencies
+          version: "latest"
+      - name: Set up Python
         run: |
-          python -m pip install --upgrade pip
-          pip install setuptools wheel build
+          uv python install 3.11
       - name: Build package
         run: |
-          python -m build  
+          uv build  
       - name: Publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1

.github/workflows/run-tests.yml

@@ -18,17 +18,19 @@ jobs:
     runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v3
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v3
+    - name: Set up UV
+      uses: astral-sh/setup-uv@v3
       with:
-        python-version: ${{ matrix.python-version }}
+        version: "latest"
+    - name: Set up Python ${{ matrix.python-version }}
+      run: |
+        uv python install ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
-        python -m pip install -e ".[dev]"
+        uv sync --dev
     - name: Type check with MyPy
       run: |
-        python -m mypy ./leetcode_study_tool
+        uv run mypy ./leetcode_study_tool
     - name: Test with pytest, ensuring 75% coverage
       run: |
-        pip install pytest pytest-cov
-        pytest tests/ --cov --cov-fail-under=75
+        uv run pytest tests/ --cov --cov-fail-under=75

.python-version

@@ -0,0 +1 @@
+3.8

Makefile

@@ -1,3 +1,5 @@
+.PHONY: all format format-check test type-check sync install clean
+
 all:	
 	@echo "Running make format"
 	$(MAKE) format
@@ -8,16 +10,25 @@ all:
 	@echo "Running make type-check"
 	$(MAKE) type-check
 
+sync:
+	uv sync --dev
+
+install:
+	uv sync
+
+clean:
+	rm -rf .venv
+
 format:
-	python -m ruff format ./leetcode_study_tool
-	python -m ruff check ./leetcode_study_tool --fix
+	uv run ruff format ./leetcode_study_tool
+	uv run ruff check ./leetcode_study_tool --fix
 
 format-check:
-	python -m ruff format ./leetcode_study_tool --check
-	python -m ruff check ./leetcode_study_tool 
+	uv run ruff format ./leetcode_study_tool --check
+	uv run ruff check ./leetcode_study_tool 
 
 test:
-	pytest tests/ --cov --cov-fail-under=85
+	uv run pytest tests/ --cov --cov-fail-under=85
 
 type-check:
-	python -m mypy ./leetcode_study_tool
+	uv run mypy ./leetcode_study_tool

README.md

@@ -154,14 +154,66 @@ Here's an example template that highlights the solution code:
 {{ data.tags|map(attribute='slug')|join(' ') }}
 ```
 
+## 🛠 Development 
+
+For developers who want to contribute to this project, we use UV for dependency management:
+
+### Setup with UV
+```shell
+# Install UV (if not already installed)
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# Clone and setup the project
+git clone https://github.com/johnsutor/leetcode-study-tool.git
+cd leetcode-study-tool
+uv sync --dev
+```
+
+### Development Commands
+```shell
+# Format code
+make format
+
+# Check formatting
+make format-check
+
+# Run tests
+make test
+
+# Type checking
+make type-check
+
+# Sync dependencies
+make sync
+
+# Install only main dependencies
+make install
+```
+
+### Using UV Directly
+```shell
+# Add a new dependency
+uv add package-name
+
+# Add a development dependency
+uv add --dev package-name
+
+# Run any command in the project environment
+uv run python leetcode_study_tool/cli.py --help
+
+# Update lockfile
+uv lock
+```
+
 ## 🛣 Roadmap 
 - [X] Use TQDM to show card generation progress
 - [X] Add support for exporting to an excel sheet
 - [X] Add support for showing neetcode solutions on the back of the card as a 
 - [X] Add support for getting the difficulty of questions 
 - [X] Add support for Jinja templating formatters 
 - [X] Add support for including NeetCode solution code
-- [ ] Add NeetCode shorts 
+- [X] Migrate to UV for faster dependency management
+- [X] Add NeetCode shorts 
 - [ ] Add support for fetching premium questions via authentification
 - [ ] Add support for importing cards into Quizlet
 - [ ] Add support for fetching questions by topic or tag 

leetcode_study_tool/formatters.py

@@ -148,7 +148,7 @@ def format_excel(url: str, slug: str, data: dict) -> List[Union[str, date]]:
         The Excel problem for the given URL and data. The problem
         is formatted as a list of strings, where each string is a
         column in the Excel file. This row will have the ordering:
-        [id, title, url, date attempted, tags, neetcode, solutions, companies]
+        [id, title, difficulty, url, date attempted, tags, video_url, short_url, solutions, companies]
     """
     row = []
     row.append(data["id"])
@@ -158,10 +158,16 @@ def format_excel(url: str, slug: str, data: dict) -> List[Union[str, date]]:
     row.append(date.today())
     row.append(", ".join([tag["name"] for tag in data["tags"]]))
     if str(data["id"]) in LEETCODE_TO_NEETCODE:
-        neetcode = LEETCODE_TO_NEETCODE[str(data["id"])]
-        row.append(neetcode["url"])
+        neetcode_data = LEETCODE_TO_NEETCODE[str(data["id"])]
+        # Add regular video URL
+        video_url = neetcode_data.get("video", {}).get("url", "")
+        row.append(video_url)
+        # Add shorts URL in new column
+        short_url = neetcode_data.get("short", {}).get("url", "")
+        row.append(short_url)
     else:
-        row.append("")
+        row.append("")  # No regular video
+        row.append("")  # No short
     row.append(data.get("neetcode_solution", ""))
     row.append(
         "\n".join(