Add IP address logging

project_amber/helpers/auth.py

@@ -94,8 +94,12 @@ def createSession(name: str, password: str) -> str:
         # have to reveal the presence or absence of a user in the system
     if verifyPassword(user.id, password):
         token = sha256(gensalt() + bytes(str(time()).encode())).hexdigest()
-        session = Session(token=token, user=user.id, login_time=time())
-        log("User %s logged in" % user.name)
+        session = Session(token=token, user=user.id, login_time=time(), \
+            address=request.remote_addr)
+        log("User {0} logged in from {1}".format(
+            user.name,
+            request.remote_addr
+        ))
         db.session.add(session)
         db.session.commit()
         return token

project_amber/models/auth.py

@@ -18,6 +18,7 @@ class Session(db.Model):
     token = db.Column(db.String(256), primary_key=True)
     user = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
     login_time = db.Column(db.Integer, nullable=False)
+    address = db.Column(db.String(100), nullable=False)
     def __repr__(self):
-        return "<Session token='%s' user='%d' login_time='%d'>" % \
-            self.token, self.user, self.login_time
+        return "<Session token='%s' user='%d' login_time='%d' ip='%s'>" % \
+            self.token, self.user, self.login_time, self.address