From 64728109d6663577e2cfc7af0a2296fd99839e56 Mon Sep 17 00:00:00 2001
From: Cullen MacDonald <cullenbmacdonald@gmail.com>
Date: Wed, 28 Jun 2017 11:34:41 -0400
Subject: [PATCH] added the ability to use channels with files

---
 slack_cleaner/cli.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/slack_cleaner/cli.py b/slack_cleaner/cli.py
index 6dbba99..126d87b 100644
--- a/slack_cleaner/cli.py
+++ b/slack_cleaner/cli.py
@@ -162,7 +162,7 @@ def get_user_name(m):
     counter.increase()
 
 
-def remove_files(time_range, user_id=None, types=None):
+def remove_files(time_range, user_id=None, types=None, channel_id=None):
     # Setup time range for query
     oldest = time_range.start_ts
     latest = time_range.end_ts
@@ -174,6 +174,7 @@ def remove_files(time_range, user_id=None, types=None):
     has_more = True
     while has_more:
         res = slack.files.list(user=user_id, ts_from=oldest, ts_to=latest,
+                               channel=channel_id,
                                types=types, page=page).body
 
         if not res['ok']:
@@ -314,6 +315,15 @@ def message_cleaner():
 def file_cleaner():
     _user_id = None
     _types = None
+    _channel_id = None
+
+    # If channel's name is supplied
+    if args.channel_name:
+        _channel_id = get_channel_id_by_name(args.channel_name)
+
+    # If group's name is supplied
+    if args.group_name:
+        _channel_id = get_group_id_by_name(args.group_name)
 
     if args.user_name:
         # A little bit tricky here, we use -1 to indicates `--user=*`
@@ -328,7 +338,11 @@ def file_cleaner():
     if args.types:
         _types = args.types
 
-    remove_files(time_range, _user_id, _types)
+    if (args.channel_name or args.group_name) and _channel_id is None:
+        sys.exit('Channel, direct message or private group not found')
+
+    remove_files(time_range, user_id=_user_id, types=_types,
+                 channel_id=_channel_id)
 
 
 def main():