Add edit button

partitioncloud/modules/albums.py

@@ -279,10 +279,10 @@ def add_partition(album_uuid):
 
             db.execute(
                 """
-                INSERT INTO partition (uuid, name, author, body)
+                INSERT INTO partition (uuid, name, author, body, user_id)
-                VALUES (?, ?, ?, ?)
+                VALUES (?, ?, ?, ?, ?)
                 """,
-                (partition_uuid, request.form["name"], author, body),
+                (partition_uuid, request.form["name"], author, body, user.id),
             )
             db.commit()
 

partitioncloud/modules/utils.py

@@ -87,7 +87,7 @@ class User():
 
 
     def get_color(self):
-        integer = int.from_bytes(self.username.encode(), "little") % 16777215
+        integer = hash(self.username) % 16777215
         return "#" + str(hex(integer))[2:]
 
 
@@ -151,7 +151,7 @@ class Album():
         db = get_db()
         return db.execute(
             """
-            SELECT partition.uuid, partition.name, partition.author FROM partition
+            SELECT partition.uuid, partition.name, partition.author, partition.user_id FROM partition
             JOIN contient_partition ON partition_uuid = partition.uuid
             JOIN album ON album.id = album_id
             WHERE album.uuid = ?

partitioncloud/schema.sql

@@ -16,7 +16,8 @@ CREATE TABLE partition (
 	uuid TEXT(36) PRIMARY KEY,
 	name TEXT NOT NULL,
 	author TEXT,
-	body TEXT
+	body TEXT,
+	user_id INTEGER
 );
 
 CREATE TABLE album (

partitioncloud/templates/albums/album.html

@@ -39,6 +39,9 @@
     <div class="partition" id="partition-{{ partition['uuid'] }}">
       <img class="partition-thumbnail" src="/static/thumbnails/{{ partition['uuid'] }}.jpg">
       <div class="partition-description">
+        {% if partition["user_id"] == g.user.id or g.user.access_level == 1 %}
+        <a href="/partition/{{ partition['uuid'] }}/edit"><div class="edit-button">✏️</div></a>
+        {% endif %}
         <div class="partition-name">{{ partition["name"] }}</div>
         <div class="partition-author">{{ partition["author"] }}</div>
       </div>