diff --git a/python-lit.spec b/python-lit.spec
index c81dced..f669671 100644
--- a/python-lit.spec
+++ b/python-lit.spec
@@ -1,21 +1,39 @@
-%global lit_version 16.0.6
-#%%global post_ver 0
+%bcond_with snapshot_build
+
+%if %{with snapshot_build}
+# Unlock LLVM Snapshot LUA functions
+%{llvm_sb_verbose}
+%{llvm_sb}
+%endif
+
+%global lit_version 17.0.1
+#global rc_ver 4
+#global post_ver 0
+
+%global python_lit_srcdir %{srcname}-%{version}%{?rc_ver:rc%{rc_ver}}%{?post_ver:.post%{post_ver}}
+
+%if %{with snapshot_build}
+%undefine rc_ver
+%global lit_version %{llvm_snapshot_version}
+%global python_lit_srcdir llvm-%{lit_version}.src/utils/lit
+%endif
 
 %bcond_without check
 
 Name: python-lit
-Version: %{lit_version}
-Release: 2%{?dist}
+Version: %{lit_version}%{?rc_ver:~rc%{rc_ver}}%{?llvm_snapshot_version_suffix:~%{llvm_snapshot_version_suffix}}
+Release: 1%{?dist}
 BuildArch: noarch
 
 License: NCSA
 Summary: Tool for executing llvm test suites
 URL: https://pypi.python.org/pypi/lit
-Source0: %{pypi_source lit %{lit_version}%{?post_ver:.post%{post_ver}}}
-# Remove the license file if it gets included in the tarball again.
-Source1: https://raw.githubusercontent.com/llvm/llvm-project/llvmorg-%{lit_version}%{?rc_ver:rc%{rc_ver}}/llvm/utils/lit/LICENSE.TXT
-
-Patch0: D149746.diff
+%if %{without snapshot_build}
+Source0: %{pypi_source lit %{lit_version}%{?rc_ver:rc%{rc_ver}}%{?post_ver:.post%{post_ver}}}
+%else
+Source0:  %{llvm_snapshot_source_prefix}llvm-%{llvm_snapshot_yyyymmdd}.src.tar.xz
+%{llvm_snapshot_extra_source_tags}
+%endif
 
 # for file check
 %if %{with check}
@@ -31,13 +49,17 @@ lit is a tool used by the LLVM project for executing its test suites.
 Summary: LLVM lit test runner for Python 3
 
 Requires: python3-setuptools
+Recommends: python3-psutil
 
 %description -n python3-lit
 lit is a tool used by the LLVM project for executing its test suites.
 
 %prep
-%autosetup -n lit-%{lit_version}%{?post_ver:.post%{post_ver}} -p4
-cp %{SOURCE1} ./
+%if %{with snapshot_build}
+%autosetup -n %{python_lit_srcdir} -p4
+%else
+%autosetup -n lit-%{lit_version}%{?rc_ver:rc%{rc_ver}}%{?post_ver:.post%{post_ver}} -p4
+%endif
 
 %build
 %py3_build
@@ -48,7 +70,7 @@ cp %{SOURCE1} ./
 # Strip out #!/usr/bin/env python
 sed -i -e '1{\@^#!/usr/bin/env python@d}' %{buildroot}%{python3_sitelib}/lit/*.py
 
-%if %{with check}
+%if %{with check} && %{without snapshot_build}
 %check
 %{__python3} lit.py -v tests
 %endif
@@ -60,6 +82,9 @@ sed -i -e '1{\@^#!/usr/bin/env python@d}' %{buildroot}%{python3_sitelib}/lit/*.p
 %{_bindir}/lit
 
 %changelog
+* Tue Sep 26 2023 Timm Bäder <tbaeder@redhat.com> - 17.0.1-1
+- Update to 17.0.1
+
 * Fri Jun 30 2023 Nikita Popov <npopov@redhat.com> - 16.0.6-2
 - Update to LLVM 16.0.6
 
diff --git a/sources b/sources
index 3ccc9e4..6de4d1a 100644
--- a/sources
+++ b/sources
@@ -1,2 +1 @@
-SHA512 (lit-16.0.6.tar.gz) = 994140c36aa8140fc28e41f47f9cb8fff0a9e6caa43ccb19e1d7b93f01a2c77b007bd85bb44ca29c47432abf3151845f2e1c5883b6f410109702bae0f975f773
-SHA512 (LICENSE.TXT) = 4a3325224ec2c86de08cf6f9320f1c95a6de061e45b53841df8fcab755d77341c27e0e064e4330d096b413c4fbb5310362b2f6c9b5c60e907bdceb2ff9b956d0
+SHA512 (lit-17.0.1.tar.gz) = 3afe4ee266ef0105658ede50e744561ee3166a8853c353d6013b7ae57ecf6c4aa8c1654f55644211cb34cbab991f849561e90c2008d8baf21d5103996523f8f2