Use single source of truth for package version

granian/__init__.py

@@ -1 +1,2 @@
+from ._granian import __version__  # noqa
 from .server import Granian  # noqa

granian/_granian.pyi

@@ -2,6 +2,8 @@ from typing import Any, Dict, List, Optional, Tuple
 
 from ._types import WebsocketMessage
 
+__version__: str
+
 class ASGIScope:
     def as_dict(self, root_path: str) -> Dict[str, Any]: ...
 

granian/cli.py

@@ -4,7 +4,7 @@
 
 import typer
 
-from .__version__ import __version__
+from . import __version__
 from .constants import HTTPModes, Interfaces, Loops, ThreadModes
 from .http import HTTP1Settings, HTTP2Settings
 from .log import LogLevels

src/lib.rs

@@ -2,6 +2,7 @@
 static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
 
 use pyo3::prelude::*;
+use std::sync::OnceLock;
 
 mod asgi;
 mod callbacks;
@@ -17,8 +18,18 @@ mod workers;
 mod ws;
 mod wsgi;
 
+pub fn get_granian_version() -> &'static str {
+    static GRANIAN_VERSION: OnceLock<String> = OnceLock::new();
+
+    GRANIAN_VERSION.get_or_init(|| {
+        let version = env!("CARGO_PKG_VERSION");
+        version.replace("-alpha", "a").replace("-beta", "b")
+    })
+}
+
 #[pymodule]
 fn _granian(py: Python, module: &PyModule) -> PyResult<()> {
+    module.add("__version__", get_granian_version())?;
     asgi::init_pymodule(module)?;
     rsgi::init_pymodule(py, module)?;
     tcp::init_pymodule(module)?;