168 lines
7.9 KiB
Python
168 lines
7.9 KiB
Python
from urllib.parse import urljoin
|
|
|
|
from decouple import config
|
|
|
|
import schemas
|
|
from chalicelib.core import sessions
|
|
from chalicelib.core.collaboration_slack import Slack
|
|
from chalicelib.utils import pg_client, helper
|
|
from chalicelib.utils.TimeUTC import TimeUTC
|
|
|
|
|
|
def get_note(tenant_id, project_id, user_id, note_id, share=None):
|
|
with pg_client.PostgresClient() as cur:
|
|
query = cur.mogrify(f"""SELECT sessions_notes.*, users.name AS creator_name
|
|
{",(SELECT name FROM users WHERE user_id=%(share)s AND deleted_at ISNULL) AS share_name" if share else ""}
|
|
FROM sessions_notes INNER JOIN users USING (user_id)
|
|
WHERE sessions_notes.project_id = %(project_id)s
|
|
AND sessions_notes.note_id = %(note_id)s
|
|
AND sessions_notes.deleted_at IS NULL
|
|
AND (sessions_notes.user_id = %(user_id)s OR sessions_notes.is_public);""",
|
|
{"project_id": project_id, "user_id": user_id, "tenant_id": tenant_id,
|
|
"note_id": note_id, "share": share})
|
|
|
|
cur.execute(query=query)
|
|
row = cur.fetchone()
|
|
row = helper.dict_to_camel_case(row)
|
|
if row:
|
|
row["createdAt"] = TimeUTC.datetime_to_timestamp(row["createdAt"])
|
|
return row
|
|
|
|
|
|
def get_session_notes(tenant_id, project_id, session_id, user_id):
|
|
with pg_client.PostgresClient() as cur:
|
|
query = cur.mogrify(f"""SELECT sessions_notes.*
|
|
FROM sessions_notes
|
|
WHERE sessions_notes.project_id = %(project_id)s
|
|
AND sessions_notes.deleted_at IS NULL
|
|
AND sessions_notes.session_id = %(session_id)s
|
|
AND (sessions_notes.user_id = %(user_id)s
|
|
OR sessions_notes.is_public)
|
|
ORDER BY created_at DESC;""",
|
|
{"project_id": project_id, "user_id": user_id,
|
|
"tenant_id": tenant_id, "session_id": session_id})
|
|
|
|
cur.execute(query=query)
|
|
rows = cur.fetchall()
|
|
rows = helper.list_to_camel_case(rows)
|
|
for row in rows:
|
|
row["createdAt"] = TimeUTC.datetime_to_timestamp(row["createdAt"])
|
|
return rows
|
|
|
|
|
|
def get_all_notes_by_project_id(tenant_id, project_id, user_id, data: schemas.SearchNoteSchema):
|
|
with pg_client.PostgresClient() as cur:
|
|
conditions = ["sessions_notes.project_id = %(project_id)s", "sessions_notes.deleted_at IS NULL"]
|
|
extra_params = {}
|
|
if data.tags and len(data.tags) > 0:
|
|
k = "tag_value"
|
|
conditions.append(
|
|
sessions._multiple_conditions(f"%({k})s = sessions_notes.tag", data.tags, value_key=k))
|
|
extra_params = sessions._multiple_values(data.tags, value_key=k)
|
|
if data.shared_only:
|
|
conditions.append("sessions_notes.is_public")
|
|
elif data.mine_only:
|
|
conditions.append("sessions_notes.user_id = %(user_id)s")
|
|
else:
|
|
conditions.append("(sessions_notes.user_id = %(user_id)s OR sessions_notes.is_public)")
|
|
query = cur.mogrify(f"""SELECT sessions_notes.*
|
|
FROM sessions_notes
|
|
WHERE {" AND ".join(conditions)}
|
|
ORDER BY created_at {data.order}
|
|
LIMIT {data.limit} OFFSET {data.limit * (data.page - 1)};""",
|
|
{"project_id": project_id, "user_id": user_id, "tenant_id": tenant_id, **extra_params})
|
|
|
|
cur.execute(query=query)
|
|
rows = cur.fetchall()
|
|
rows = helper.list_to_camel_case(rows)
|
|
for row in rows:
|
|
row["createdAt"] = TimeUTC.datetime_to_timestamp(row["createdAt"])
|
|
return rows
|
|
|
|
|
|
def create(tenant_id, user_id, project_id, session_id, data: schemas.SessionNoteSchema):
|
|
with pg_client.PostgresClient() as cur:
|
|
query = cur.mogrify(f"""INSERT INTO public.sessions_notes (message, user_id, tag, session_id, project_id, timestamp, is_public)
|
|
VALUES (%(message)s, %(user_id)s, %(tag)s, %(session_id)s, %(project_id)s, %(timestamp)s, %(is_public)s)
|
|
RETURNING *;""",
|
|
{"user_id": user_id, "project_id": project_id, "session_id": session_id, **data.dict()})
|
|
cur.execute(query)
|
|
result = helper.dict_to_camel_case(cur.fetchone())
|
|
if result:
|
|
result["createdAt"] = TimeUTC.datetime_to_timestamp(result["createdAt"])
|
|
return result
|
|
|
|
|
|
def edit(tenant_id, user_id, project_id, note_id, data: schemas.SessionUpdateNoteSchema):
|
|
sub_query = []
|
|
if data.message is not None:
|
|
sub_query.append("message = %(message)s")
|
|
if data.tag is not None and len(data.tag) > 0:
|
|
sub_query.append("tag = %(tag)s")
|
|
if data.is_public is not None:
|
|
sub_query.append("is_public = %(is_public)s")
|
|
if data.timestamp is not None:
|
|
sub_query.append("timestamp = %(timestamp)s")
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
cur.mogrify(f"""UPDATE public.sessions_notes
|
|
SET
|
|
{" ,".join(sub_query)}
|
|
WHERE
|
|
project_id = %(project_id)s
|
|
AND user_id = %(user_id)s
|
|
AND note_id = %(note_id)s
|
|
AND deleted_at ISNULL
|
|
RETURNING *;""",
|
|
{"project_id": project_id, "user_id": user_id, "note_id": note_id, **data.dict()})
|
|
)
|
|
row = helper.dict_to_camel_case(cur.fetchone())
|
|
if row:
|
|
row["createdAt"] = TimeUTC.datetime_to_timestamp(row["createdAt"])
|
|
return row
|
|
|
|
|
|
def delete(tenant_id, user_id, project_id, note_id):
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
cur.mogrify(""" UPDATE public.sessions_notes
|
|
SET deleted_at = timezone('utc'::text, now())
|
|
WHERE note_id = %(note_id)s
|
|
AND project_id = %(project_id)s
|
|
AND user_id = %(user_id)s
|
|
AND deleted_at ISNULL;""",
|
|
{"project_id": project_id, "user_id": user_id, "note_id": note_id})
|
|
)
|
|
return {"data": {"state": "success"}}
|
|
|
|
|
|
def share_to_slack(tenant_id, user_id, project_id, note_id, webhook_id):
|
|
note = get_note(tenant_id=tenant_id, project_id=project_id, user_id=user_id, note_id=note_id, share=user_id)
|
|
if note is None:
|
|
return {"errors": ["Note not found"]}
|
|
session_url = urljoin(config('SITE_URL'), f"{note['projectId']}/session/{note['sessionId']}?note={note['noteId']}")
|
|
if note["timestamp"] > 0:
|
|
session_url += f"&jumpto={note['timestamp']}"
|
|
title = f"<{session_url}|Note for session {note['sessionId']}>"
|
|
|
|
blocks = [{"type": "section",
|
|
"fields": [{"type": "mrkdwn",
|
|
"text": title}]},
|
|
{"type": "section",
|
|
"fields": [{"type": "plain_text",
|
|
"text": note["message"]}]}]
|
|
if note["tag"]:
|
|
blocks.append({"type": "context",
|
|
"elements": [{"type": "plain_text",
|
|
"text": f"Tag: *{note['tag']}*"}]})
|
|
bottom = f"Created by {note['creatorName'].capitalize()}"
|
|
if user_id != note["userId"]:
|
|
bottom += f"\nSent by {note['shareName']}: "
|
|
blocks.append({"type": "context",
|
|
"elements": [{"type": "plain_text",
|
|
"text": bottom}]})
|
|
return Slack.send_raw(
|
|
tenant_id=tenant_id,
|
|
webhook_id=webhook_id,
|
|
body={"blocks": blocks}
|
|
)
|