Add an Unrecognized variant to PySourceType

crates/ruff_cli/src/commands/add_noqa.rs

@@ -46,8 +46,9 @@ pub(crate) fn add_noqa(
         .flatten()
         .filter_map(|entry| {
             let path = entry.path();
-            let SourceType::Python(source_type @ (PySourceType::Python | PySourceType::Stub)) =
+            let SourceType::Python(
-                SourceType::from(path)
+                source_type @ (PySourceType::Py | PySourceType::Pyi | PySourceType::Pyw),
+            ) = SourceType::from(path)
             else {
                 return None;
             };

crates/ruff_cli/src/commands/format.rs

@@ -42,7 +42,7 @@ pub(crate) fn format(cli: &Arguments, overrides: &Overrides) -> Result<ExitStatu
             let path = entry.path();
             if matches!(
                 SourceType::from(path),
-                SourceType::Python(PySourceType::Python | PySourceType::Stub)
+                SourceType::Python(PySourceType::Py | PySourceType::Pyi | PySourceType::Pyw)
             ) {
                 let line_length = resolver.resolve(path, &pyproject_config).line_length;
                 let options = PyFormatOptions::from_extension(path)

crates/ruff_python_ast/src/lib.rs

@@ -35,7 +35,7 @@ pub enum SourceType {
 
 impl Default for SourceType {
     fn default() -> Self {
-        Self::Python(PySourceType::Python)
+        Self::Python(PySourceType::Py)
     }
 }
 
@@ -70,20 +70,32 @@ pub enum TomlSourceType {
 pub enum PySourceType {
     /// The source is a Python file (`.py`).
     #[default]
-    Python,
+    Py,
     /// The source is a Python stub file (`.pyi`).
-    Stub,
+    Pyi,
+    /// The source is a Python script with a graphical user interface (`.pyw`).
+    Pyw,
     /// The source is a Jupyter notebook (`.ipynb`).
     Ipynb,
+    /// The source is an unrecognized Python file.
+    Unrecognized,
+}
+
+impl PySourceType {
+    /// Return `true` if the source type is a stub file.
+    pub const fn is_stub(self) -> bool {
+        matches!(self, Self::Pyi)
+    }
 }
 
 impl From<&Path> for PySourceType {
     fn from(path: &Path) -> Self {
         match path.extension() {
-            Some(ext) if ext == "py" => PySourceType::Python,
+            Some(ext) if ext == "py" => PySourceType::Py,
-            Some(ext) if ext == "pyi" => PySourceType::Stub,
+            Some(ext) if ext == "pyi" => PySourceType::Pyi,
+            Some(ext) if ext == "pyw" => PySourceType::Pyw,
             Some(ext) if ext == "ipynb" => PySourceType::Ipynb,
-            _ => PySourceType::Python,
+            _ => PySourceType::Py,
         }
     }
 }

crates/ruff_python_parser/src/lib.rs

@@ -312,7 +312,10 @@ pub trait AsMode {
 impl AsMode for PySourceType {
     fn as_mode(&self) -> Mode {
         match self {
-            PySourceType::Python | PySourceType::Stub => Mode::Module,
+            PySourceType::Py
+            | PySourceType::Pyi
+            | PySourceType::Pyw
+            | PySourceType::Unrecognized => Mode::Module,
             PySourceType::Ipynb => Mode::Jupyter,
         }
     }