-
Notifications
You must be signed in to change notification settings - Fork 0
/
data_sync.py
74 lines (53 loc) · 1.93 KB
/
data_sync.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import os
import click
def mkdir_baiduyun_dir(baiduyun_dir):
"""Create a directory in the baiduyun directory"""
os.system("baidupcs-go mkdir %s" % baiduyun_dir)
def del_baiduyun_dir(baiduyun_dir):
"""Delete the directory in the baiduyun directory"""
os.system("baidupcs-go rm %s/data" % baiduyun_dir)
def upload_baiduyun_dir(local_dir, baiduyun_dir):
"""Upload the directory to the baiduyun directory"""
os.system("baidupcs-go upload %s %s" % (local_dir, baiduyun_dir))
def download_baiduyun_dir(baiduyun_dir):
"""Download the directory from the baiduyun directory"""
os.system("baidupcs-go download %s/data ./" % baiduyun_dir)
@click.group()
def data_sync():
pass
@click.command()
@click.option(
"--baiduyun_dir",
default="/data/Comparative-Study-of-Methods-for-Drive-Time-Estimation",
help="Init the baiduyun_dir",
)
def init(baiduyun_dir):
"""Create a directory in the baiduyun directory"""
mkdir_baiduyun_dir("/data/%s/data" % baiduyun_dir)
@click.command()
@click.option("--local_dir", default="./data", help="The local directory")
@click.option(
"--baiduyun_dir",
default="/data/Comparative-Study-of-Methods-for-Drive-Time-Estimation",
help="The baiduyun directory",
)
def push_all(local_dir, baiduyun_dir):
"""Synchronize data to the baiduyun directory"""
# mkdir_baiduyun_dir(baiduyun_dir)
del_baiduyun_dir(baiduyun_dir)
upload_baiduyun_dir(local_dir, baiduyun_dir)
# download_baiduyun_dir()
@click.command()
@click.option("--local_dir", default="./data", help="The local directory")
@click.option(
"--baiduyun_dir",
default="/data/Comparative-Study-of-Methods-for-Drive-Time-Estimation",
help="The baiduyun directory",
)
def pull_all(baiduyun_dir):
"""Synchronize data from the baiduyun directory"""
download_baiduyun_dir(baiduyun_dir)
data_sync.add_command(push_all)
data_sync.add_command(pull_all)
if __name__ == "__main__":
data_sync()