用于复制随机文件夹的脚本

Question

我想实现一个我一直在想的小项目。我的桌面上有一个庞大的音乐库，我正在寻找一种方法/脚本来选择随机数量的文件夹（比如 25 个专辑），如果可行，将它们复制到我的汽车的 USB 驱动器上。我的完整愿景是一个能够执行以下步骤的脚本：

1. 擦除 USB 驱动器（它将仅用于音乐）
2. 专辑随机选择（数据量上限？固定数量？）
3. 将选择复制到 USB 驱动器

这可以通过简单的脚本实现吗？我记得一些音乐组织者有这个选项，但我正在寻找一种更简单的无头服务器。

Answer 1

下面的脚本应该可以完成这项工作：

#!/usr/bin/env python3
import random
import os
import subprocess
import shutil

# set the desired number of folders to pick below
n_selection = 5
# set the name of the flash drive below
flashdr = "Lexar"
# set the source directory (with media folders) below
sourcedr = "/path/to/mediafiles"
# ---

try:
    targetdr = [l.split("part ")[-1] for l in subprocess.check_output("lsblk")\
                .decode("utf-8").splitlines()if l.endswith(flashdr)][0]
except IndexError:
    pass
else:
    # empty the flash drive
    for item in os.listdir(targetdr):
        obj = os.path.join(targetdr, item)
        try:
            shutil.rmtree(obj)
        except NotADirectoryError:
            os.remove(obj)
    # list the source dirs
    srclist = []
    for dr in  os.listdir(sourcedr):
        fullpath = os.path.join(sourcedr, dr)
        if os.path.isdir(fullpath):
            srclist.append([dr, fullpath])
    # copy the files
    for picked in random.sample(srclist, n_selection):
        shutil.copytree(picked[1], os.path.join(targetdr, picked[0]))
        srclist.remove(picked)

它将目录从源目录的第一个子级别复制到目标闪存驱动器中。

这似乎是最有意义的，因为随机递归复制文件夹会导致文件夹大小和数量子级别的巨大差异。尽管如此，我还是在这个答案的底部添加了它作为第二个选项。

如何使用

1. 将脚本复制到一个空文件中，另存为 create_mediausb.py
2. 在 head 部分，设置要选择的文件数量、闪存驱动器的名称（脚本将找到其路径）和文件夹的源目录。

3. 使用以下命令运行脚本：

   python3 /path/to/create_mediausb.py 
   

4. 如果一切正常，请将其添加到快捷键：选择：系统设置>“键盘”>“快捷方式”>“自定义快捷方式”。单击“+”并添加命令：

   python3 /path/to/create_mediausb.py 
   

第二个选项：递归选择

python3 /path/to/create_mediausb.py 

剧本：

1. 查找闪存驱动器的路径：

   python3 /path/to/create_mediausb.py 
   

   如果未找到闪存驱动器，则到此结束

2. 如果找到驱动器，则将其清空

   #!/usr/bin/env python3
   import random
   import os
   import subprocess
   import shutil
   
   n_selection = 5
   flashdr = "Lexar"
   sourcedr = "/home/jacob/Bureaublad/GW_site_nafestival_2015/pix/general"
   
   try:
       targetdr = [l.split("part ")[-1] for l in subprocess.check_output("lsblk")\
                   .decode("utf-8").splitlines()if l.endswith(flashdr)][0]
   except IndexError:
       pass
   else:
       # empty the flash drive
       for item in os.listdir(targetdr):
           obj = os.path.join(targetdr, item)
           try:
               shutil.rmtree(obj)
           except NotADirectoryError:
               os.remove(obj)
       # list the source dirs
       srclist = []
       for root, dirs, files in os.walk(sourcedr):
           for dr in dirs:
               srclist.append([dr, os.path.join(root, dr)])
       # copy the files
       for picked in random.sample(srclist, n_selection):
           shutil.copytree(picked[1], os.path.join(targetdr, picked[0]))
           srclist.remove(picked)
   

3. 然后，由于我们需要整个目录列表随机进行适当的选择，因此我们在进行选择之前创建一个列表：

   try:
       targetdr = [l.split("part ")[-1] for l in subprocess.check_output("lsblk")\
                   .decode("utf-8").splitlines()if l.endswith(flashdr)][0]
   except IndexError:
       pass
   

4. 最有趣的部分是进行选择。我们随机选择一个目录，将其从列表中删除以防止重复选择，然后随机选择另一个目录，将其删除，依此类推，直到达到所需选择的数量：

   # empty the flash drive
   for item in os.listdir(targetdr):
       obj = (targetdr+"/"+item)
       try:
           shutil.rmtree(obj)
       except NotADirectoryError:
           os.remove(obj)
   

Answer 2

你可以使用 find 和 shuf：

#!/bin/bash

SOURCE="path/to/source"
DESTINATION="path/to/destination"
COUNT=25

rm -r "${DESTINATION}/"*
find "$SOURCE" -mindepth 2 -maxdepth 2 -type d|shuf -n $COUNT|xargs -d'\n' -I{} cp -r "{}" "$DESTINATION"

Answer 3

我用Python编写了一个脚本：

该脚本位于 GitHub Gist。（下载）

用法：

python3 RandomCopier.py [source folder] [destination folder] [number to copy]

复制方法：

注意：它不会直接复制源文件夹中的任何文件，只会复制其子文件夹中的文件。

比如说，源文件夹src是：

src
|- a
|  |- file_a
|  |- file_a_2
|
|- b
|  |- file_b
|
|- c
|  |- file_c
|
|- file_src

然后，目标文件夹dest将是随机复制的 2 个文件夹，例如：

dest
|- a
|  |- file_a
|  |- file_a_2
|
|- c
|  |- file_c