Merge pull request #2 from masenf/user-rename

rename User to LocalUser

Author: masenf

.github/workflows/publish.yml

@@ -0,0 +1,21 @@
+name: Publish Component to PyPI
+
+on:
+  push:
+    tags:
+     - '*'
+
+jobs:
+  publish:
+    name: Publish Component to PyPI
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@master
+    - name: Set up Python 3.12
+      uses: actions/setup-python@v3
+      with:
+        python-version: '3.12'
+    - name: Install package
+      run: pip install .
+    - name: Publish to PyPI
+      run: reflex component publish -t ${{ secrets.PYPI_TOKEN }} --no-share --no-validate-project-info

README.md

@@ -90,7 +90,7 @@ def protected_page():
 
 ## Customization
 
-The basic `reflex_local_auth.User` model provides password hashing and
+The basic `reflex_local_auth.LocalUser` model provides password hashing and
 verification, and an enabled flag. Additional functionality can be added by
 creating a new `UserInfo` model and creating a foreign key relationship to the
 `user.id` field.
@@ -146,7 +146,7 @@ def register_error() -> rx.Component:
         reflex_local_auth.RegistrationState.error_message != "",
         rx.callout(
             reflex_local_auth.RegistrationState.error_message,
-            icon="alert_triangle",
+            icon="triangle_alert",
             color_scheme="red",
             role="alert",
             width="100%",
@@ -234,4 +234,21 @@ def user_info():
             ),
         ),
     )
+```
+
+## Migrating from 0.0.x to 0.1.x
+
+The `User` model has been renamed to `LocalUser` and the `AuthSession` model has
+been renamed to `LocalAuthSession`. If your app was using reflex-local-auth 0.0.x,
+then you will need to make manual changes to migration script to copy existing user
+data into the new tables _after_ running `reflex db makemigrations`.
+
+See [`local_auth_demo/alembic/version/cb01e050df85_.py`](local_auth_demo/alembic/version/cb01e050df85_.py) for an example migration script.
+
+Importantly, your `upgrade` function should include the following lines, after creating
+the new tables and before dropping the old tables:
+
+```python
+    op.execute("INSERT INTO localuser SELECT * FROM user;")
+    op.execute("INSERT INTO localauthsession SELECT * FROM authsession;")
 ```

custom_components/reflex_local_auth/__init__.py

@@ -4,13 +4,13 @@
 from .login import require_login, LoginState
 from .registration import RegistrationState
 from .routes import set_login_route, set_register_route
-from .user import User
+from .user import LocalUser
 
 __all__ = [
     "LocalAuthState",
+    "LocalUser",
     "LoginState",
     "RegistrationState",
-    "User",
     "pages",
     "routes",
     "require_login",

custom_components/reflex_local_auth/auth_session.py

@@ -5,7 +5,7 @@
 import reflex as rx
 
 
-class AuthSession(
+class LocalAuthSession(
     rx.Model,
     table=True,  # type: ignore
 ):

custom_components/reflex_local_auth/local_auth.py

@@ -12,8 +12,8 @@
 
 import reflex as rx
 
-from .auth_session import AuthSession
-from .user import User
+from .auth_session import LocalAuthSession
+from .user import LocalUser
 
 
 AUTH_TOKEN_LOCAL_STORAGE_KEY = "_auth_token"
@@ -25,26 +25,26 @@ class LocalAuthState(rx.State):
     auth_token: str = rx.LocalStorage(name=AUTH_TOKEN_LOCAL_STORAGE_KEY)
 
     @rx.cached_var
-    def authenticated_user(self) -> User:
+    def authenticated_user(self) -> LocalUser:
         """The currently authenticated user, or a dummy user if not authenticated.
 
         Returns:
-            A User instance with id=-1 if not authenticated, or the User instance
+            A LocalUser instance with id=-1 if not authenticated, or the LocalUser instance
             corresponding to the currently authenticated user.
         """
         with rx.session() as session:
             result = session.exec(
-                select(User, AuthSession).where(
-                    AuthSession.session_id == self.auth_token,
-                    AuthSession.expiration
+                select(LocalUser, LocalAuthSession).where(
+                    LocalAuthSession.session_id == self.auth_token,
+                    LocalAuthSession.expiration
                     >= datetime.datetime.now(datetime.timezone.utc),
-                    User.id == AuthSession.user_id,
+                    LocalUser.id == LocalAuthSession.user_id,
                 ),
             ).first()
             if result:
                 user, session = result
                 return user
-        return User(id=-1)  # type: ignore
+        return LocalUser(id=-1)  # type: ignore
 
     @rx.cached_var
     def is_authenticated(self) -> bool:
@@ -56,10 +56,10 @@ def is_authenticated(self) -> bool:
         return self.authenticated_user.id >= 0
 
     def do_logout(self) -> None:
-        """Destroy AuthSessions associated with the auth_token."""
+        """Destroy LocalAuthSessions associated with the auth_token."""
         with rx.session() as session:
             for auth_session in session.exec(
-                select(AuthSession).where(AuthSession.session_id == self.auth_token)
+                select(LocalAuthSession).where(LocalAuthSession.session_id == self.auth_token)
             ).all():
                 session.delete(auth_session)
             session.commit()
@@ -70,14 +70,14 @@ def _login(
         user_id: int,
         expiration_delta: datetime.timedelta = DEFAULT_AUTH_SESSION_EXPIRATION_DELTA,
     ) -> None:
-        """Create an AuthSession for the given user_id.
+        """Create an LocalAuthSession for the given user_id.
 
-        If the auth_token is already associated with an AuthSession, it will be
+        If the auth_token is already associated with an LocalAuthSession, it will be
         logged out first.
 
         Args:
-            user_id: The user ID to associate with the AuthSession.
-            expiration_delta: The amount of time before the AuthSession expires.
+            user_id: The user ID to associate with the LocalAuthSession.
+            expiration_delta: The amount of time before the LocalAuthSession expires.
         """
         if self.is_authenticated:
             self.do_logout()
@@ -86,7 +86,7 @@ def _login(
         self.auth_token = self.auth_token or self.router.session.client_token
         with rx.session() as session:
             session.add(
-                AuthSession(  # type: ignore
+                LocalAuthSession(  # type: ignore
                     user_id=user_id,
                     session_id=self.auth_token,
                     expiration=datetime.datetime.now(datetime.timezone.utc)

custom_components/reflex_local_auth/login.py

@@ -6,7 +6,7 @@
 
 from . import routes
 from .local_auth import LocalAuthState
-from .user import User
+from .user import LocalUser
 
 
 class LoginState(LocalAuthState):
@@ -26,7 +26,7 @@ def on_submit(self, form_data) -> rx.event.EventSpec:
         password = form_data["password"]
         with rx.session() as session:
             user = session.exec(
-                select(User).where(User.username == username)
+                select(LocalUser).where(LocalUser.username == username)
             ).one_or_none()
         if user is not None and not user.enabled:
             self.error_message = "This account is disabled."

custom_components/reflex_local_auth/pages/login.py

@@ -19,7 +19,7 @@ def login_error() -> rx.Component:
         LoginState.error_message != "",
         rx.callout(
             LoginState.error_message,
-            icon="alert_triangle",
+            icon="triangle_alert",
             color_scheme="red",
             role="alert",
             width="100%",

custom_components/reflex_local_auth/pages/registration.py

@@ -19,7 +19,7 @@ def register_error() -> rx.Component:
         RegistrationState.error_message != "",
         rx.callout(
             RegistrationState.error_message,
-            icon="alert_triangle",
+            icon="triangle_alert",
             color_scheme="red",
             role="alert",
             width="100%",

custom_components/reflex_local_auth/registration.py

@@ -9,7 +9,7 @@
 
 from . import routes
 from .local_auth import LocalAuthState
-from .user import User
+from .user import LocalUser
 
 
 POST_REGISTRATION_DELAY = 0.5
@@ -30,7 +30,7 @@ def _validate_fields(
             return rx.set_focus("username")
         with rx.session() as session:
             existing_user = session.exec(
-                select(User).where(User.username == username)
+                select(LocalUser).where(LocalUser.username == username)
             ).one_or_none()
         if existing_user is not None:
             self.error_message = (
@@ -50,9 +50,9 @@ def _validate_fields(
     def _register_user(self, username, password) -> None:
         with rx.session() as session:
             # Create the new user and add it to the database.
-            new_user = User()  # type: ignore
+            new_user = LocalUser()  # type: ignore
             new_user.username = username
-            new_user.password_hash = User.hash_password(password)
+            new_user.password_hash = LocalUser.hash_password(password)
             new_user.enabled = True
             session.add(new_user)
             session.commit()

custom_components/reflex_local_auth/user.py

@@ -6,7 +6,7 @@
 import reflex as rx
 
 
-class User(
+class LocalUser(
     rx.Model,
     table=True,  # type: ignore
 ):