Corin1998 commited on
Commit
f03ceb1
·
verified ·
1 Parent(s): bd59e2c

Update app/tasks.py

Browse files
Files changed (1) hide show
  1. app/tasks.py +3 -5
app/tasks.py CHANGED
@@ -1,5 +1,5 @@
1
  from app.celery_app import celery
2
- from app.datebase import SessionLocal
3
  from app import models
4
  from app.llm import generate_draft
5
  from app.utils.emailer import send_email
@@ -28,7 +28,6 @@ def task_generate_draft(draft_id: int):
28
  finally:
29
  db.close()
30
 
31
-
32
  @celery.task()
33
  def task_deliver(draft_id: int):
34
  db = SessionLocal()
@@ -46,7 +45,7 @@ def task_deliver(draft_id: int):
46
  results.append({"email": res})
47
  # X
48
  if draft.deliver_x:
49
- text = f"{draft.title}\n\n{(draft.body_md or '')[:220]}" # 簡易
50
  res = post_to_x(text)
51
  delivery = models.Delivery(draft_id=draft.id, channel=models.DeliveryChannel.x, payload={"text": text[:280]}, result=res)
52
  db.add(delivery); db.commit(); db.refresh(delivery)
@@ -58,9 +57,8 @@ def task_deliver(draft_id: int):
58
  db.add(delivery); db.commit(); db.refresh(delivery)
59
  results.append({"note": res})
60
 
61
-
62
  draft.status = models.DraftStatus.sent
63
  db.add(draft); db.commit(); db.refresh(draft)
64
  return {"id": draft.id, "results": results}
65
  finally:
66
- db.close()
 
1
  from app.celery_app import celery
2
+ from app.database import SessionLocal
3
  from app import models
4
  from app.llm import generate_draft
5
  from app.utils.emailer import send_email
 
28
  finally:
29
  db.close()
30
 
 
31
  @celery.task()
32
  def task_deliver(draft_id: int):
33
  db = SessionLocal()
 
45
  results.append({"email": res})
46
  # X
47
  if draft.deliver_x:
48
+ text = f"{draft.title}\n\n{(draft.body_md or '')[:220]}"
49
  res = post_to_x(text)
50
  delivery = models.Delivery(draft_id=draft.id, channel=models.DeliveryChannel.x, payload={"text": text[:280]}, result=res)
51
  db.add(delivery); db.commit(); db.refresh(delivery)
 
57
  db.add(delivery); db.commit(); db.refresh(delivery)
58
  results.append({"note": res})
59
 
 
60
  draft.status = models.DraftStatus.sent
61
  db.add(draft); db.commit(); db.refresh(draft)
62
  return {"id": draft.id, "results": results}
63
  finally:
64
+ db.close()