[DRAFT]Sketch delete poc

cli_client/python/timesketch_cli_client/commands/sketch.py

@@ -158,3 +158,38 @@ def unarchive_sketch(ctx):
     if sketch.is_archived():
         sketch.unarchive()
         click.echo("Sketch unarchived")
+
+
+@sketch_group.command("delete", help="Delete a sketch")
+@click.option(
+    "--execute",
+    required=False,
+    is_flag=True,
+    help="Only execute the deletion if this is set.",
+)
+@click.pass_context
+def delete_sketch(ctx, execute):
+    """Delete a sketch."""
+    sketch = ctx.obj.sketch
+
+    # Dryrun:
+    click.echo("Would delete the following things")
+    click.echo(
+        f"Sketch: {sketch.id} {sketch.name} {sketch.description} {sketch.status} Labels: {sketch.labels}"  # pylint: disable=line-too-long
+    )
+
+    for timeline in sketch.list_timelines():
+        click.echo(
+            f"  Timeline: {timeline.id} {timeline.name} {timeline.description} {timeline.status}"  # pylint: disable=line-too-long
+        )
+
+    # for story in sketch.stories:
+    #    click.echo(
+    #        f"  Story: {story.id} {story.title} {story.description} {story.status} {story.created_at} {story.updated_at}" # pylint: disable=line-too-long
+    #    )
+
+    if execute:
+        click.echo("Will delete for real")
+
+        sketch.delete()
+        click.echo("Sketch deleted")

timesketch/api/v1/resources/sketch.py

@@ -495,7 +495,20 @@ def delete(self, sketch_id):
                     HTTP_STATUS_CODE_FORBIDDEN,
                     "Sketch with the label [{0:s}] cannot be deleted.".format(label),
                 )
+
         sketch.set_status(status="deleted")
+        # TODO: actually delete the sketch
+
+        for timeline in sketch.active_timelines:
+            timeline.set_status(status="deleted")
+            searchindex = timeline.searchindex
+            db_session.delete(searchindex)
+            db_session.delete(timeline)
+            # remove the opensearch index
+            self.datastore.client.indices.delete(index=searchindex.index_name)
+
+        db_session.delete(sketch)
+        db_session.commit()
         return HTTP_STATUS_CODE_OK
 
     @login_required

timesketch/models/sketch.py

@@ -58,23 +58,96 @@ class Sketch(AccessControlMixin, LabelMixin, StatusMixin, CommentMixin, BaseMode
     name = Column(Unicode(255))
     description = Column(UnicodeText())
     user_id = Column(Integer, ForeignKey("user.id"))
-    timelines = relationship("Timeline", backref="sketch", lazy="select")
-    views = relationship("View", backref="sketch", lazy="select")
-    events = relationship("Event", backref="sketch", lazy="select")
-    stories = relationship("Story", backref="sketch", lazy="select")
-    aggregations = relationship("Aggregation", backref="sketch", lazy="select")
-    attributes = relationship("Attribute", backref="sketch", lazy="select")
-    graphs = relationship("Graph", backref="sketch", lazy="select")
-    graphcaches = relationship("GraphCache", backref="sketch", lazy="select")
+    timelines = relationship(
+        "Timeline",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    views = relationship(
+        "View",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    events = relationship(
+        "Event",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    stories = relationship(
+        "Story",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    aggregations = relationship(
+        "Aggregation",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    attributes = relationship(
+        "Attribute",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    graphs = relationship(
+        "Graph",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    graphcaches = relationship(
+        "GraphCache",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
     aggregationgroups = relationship(
-        "AggregationGroup", backref="sketch", lazy="select"
-    )
-    analysis = relationship("Analysis", backref="sketch", lazy="select")
-    analysissessions = relationship("AnalysisSession", backref="sketch", lazy="select")
-    searchhistories = relationship("SearchHistory", backref="sketch", lazy="dynamic")
-    scenarios = relationship("Scenario", backref="sketch", lazy="dynamic")
-    facets = relationship("Facet", backref="sketch", lazy="dynamic")
-    questions = relationship("InvestigativeQuestion", backref="sketch", lazy="dynamic")
+        "AggregationGroup",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    analysis = relationship(
+        "Analysis",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    analysissessions = relationship(
+        "AnalysisSession",
+        backref="sketch",
+        lazy="select",
+        cascade="save-update, merge, delete",
+    )
+    searchhistories = relationship(
+        "SearchHistory",
+        backref="sketch",
+        lazy="dynamic",
+        cascade="save-update, merge, delete",
+    )
+    scenarios = relationship(
+        "Scenario",
+        backref="sketch",
+        lazy="dynamic",
+        cascade="save-update, merge, delete",
+    )
+    facets = relationship(
+        "Facet",
+        backref="sketch",
+        lazy="dynamic",
+        cascade="save-update, merge, delete",
+    )
+    questions = relationship(
+        "InvestigativeQuestion",
+        backref="sketch",
+        lazy="dynamic",
+        cascade="save-update, merge, delete",
+    )
 
     @property
     def get_named_aggregations(self):

timesketch/tsctl.py

@@ -545,6 +545,31 @@ def print_table(table_data):
         print()
 
 
+@cli.command(name="sketch-delete")
+@click.argument("sketch_id")
+def sketch_delete(sketch_id):
+    """Delete a sketch."""
+    sketch = Sketch.query.filter_by(id=sketch_id).first()
+    if not sketch:
+        print("Sketch does not exist.")
+        return
+    elif sketch.get_status.status == "deleted":
+        print(f"Sketch {sketch_id} is already deleted.")
+        return
+    else:
+        sketch_labels = [label.label for label in sketch.labels]
+        # do not delete if Sketch has a lithold label
+        if "lithold" in sketch_labels:
+            print("Sketch has lithold label, skipping")
+            return
+
+        print(f"Sketch {sketch_id} Name: ({sketch.name})")
+        sketch.delete()
+        db_session.delete(sketch)
+        db_session.commit()
+        print(f"Sketch {sketch_id} deleted.")
+
+
 @cli.command(name="sketch-info")
 @click.argument("sketch_id")
 def sketch_info(sketch_id):