Skip to content

Commit af7b75b

Browse files
chrisgorgorysteboe
authored andcommitted
addded command line flag
1 parent cdb422a commit af7b75b

File tree

2 files changed

+17
-12
lines changed

2 files changed

+17
-12
lines changed

kaggle/api/kaggle_api_extended.py

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1208,7 +1208,8 @@ def dataset_create_version(self,
12081208
version_notes,
12091209
quiet=False,
12101210
convert_to_csv=True,
1211-
delete_old_versions=False):
1211+
delete_old_versions=False,
1212+
dir_mode='skip'):
12121213
""" create a version of a dataset
12131214
12141215
Parameters
@@ -1251,7 +1252,7 @@ def dataset_create_version(self,
12511252
convert_to_csv=convert_to_csv,
12521253
category_ids=keywords,
12531254
delete_old_versions=delete_old_versions)
1254-
self.upload_files(request, resources, folder, quiet)
1255+
self.upload_files(request, resources, folder, quiet, dir_mode)
12551256

12561257
if id_no:
12571258
result = DatasetNewVersionResponse(
@@ -1280,7 +1281,8 @@ def dataset_create_version_cli(self,
12801281
version_notes,
12811282
quiet=False,
12821283
convert_to_csv=True,
1283-
delete_old_versions=False):
1284+
delete_old_versions=False,
1285+
dir_mode='skip'):
12841286
""" client wrapper for creating a version of a dataset
12851287
Parameters
12861288
==========
@@ -1296,7 +1298,8 @@ def dataset_create_version_cli(self,
12961298
version_notes,
12971299
quiet=quiet,
12981300
convert_to_csv=convert_to_csv,
1299-
delete_old_versions=delete_old_versions)
1301+
delete_old_versions=delete_old_versions,
1302+
dir_mode=dir_mode)
13001303
if result.invalidTags:
13011304
print(
13021305
('The following are not valid tags and could not be added to '
@@ -1344,7 +1347,8 @@ def dataset_create_new(self,
13441347
folder,
13451348
public=False,
13461349
quiet=False,
1347-
convert_to_csv=True):
1350+
convert_to_csv=True,
1351+
dir_mode='skip'):
13481352
""" create a new dataset, meaning the same as creating a version but
13491353
with extra metadata like license and user/owner.
13501354
Parameters
@@ -1422,7 +1426,8 @@ def dataset_create_new_cli(self,
14221426
folder=None,
14231427
public=False,
14241428
quiet=False,
1425-
convert_to_csv=True):
1429+
convert_to_csv=True,
1430+
dir_mode='skip'):
14261431
""" client wrapper for creating a new dataset
14271432
Parameters
14281433
==========
@@ -1432,7 +1437,7 @@ def dataset_create_new_cli(self,
14321437
convert_to_csv: if True, convert data to comma separated value
14331438
"""
14341439
folder = folder or os.getcwd()
1435-
result = self.dataset_create_new(folder, public, quiet, convert_to_csv)
1440+
result = self.dataset_create_new(folder, public, quiet, convert_to_csv, dir_mode)
14361441
if result.invalidTags:
14371442
print('The following are not valid tags and could not be added to '
14381443
'the dataset: ' + str(result.invalidTags))
@@ -2212,7 +2217,7 @@ def is_up_to_date(self, server_version):
22122217

22132218
return True
22142219

2215-
def upload_files(self, request, resources, folder, quiet=False):
2220+
def upload_files(self, request, resources, folder, quiet=False, dir_mode='skip'):
22162221
""" upload files in a folder
22172222
Parameters
22182223
==========
@@ -2232,7 +2237,7 @@ def upload_files(self, request, resources, folder, quiet=False):
22322237
exitcode = self._upload_file(file_name, full_path, quiet, request, resources)
22332238
if exitcode:
22342239
return
2235-
if os.path.isdir(full_path):
2240+
if os.path.isdir(full_path) and dir_mode == 'zip':
22362241
temp_dir = tempfile.mkdtemp()
22372242
try:
22382243
_, dir_name = os.path.split(full_path)

kaggle/cli.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ def parse_datasets(subparsers):
500500
action='store_false',
501501
help=Help.param_keep_tabular)
502502
parser_datasets_create_optional.add_argument(
503-
'-d',
503+
'-r',
504504
'--dir-mode',
505505
dest='dir_mode',
506506
choices=['skip', 'zip'],
@@ -543,8 +543,8 @@ def parse_datasets(subparsers):
543543
dest='convert_to_csv',
544544
action='store_false',
545545
help=Help.param_keep_tabular)
546-
parser_datasets_create_optional.add_argument(
547-
'-d',
546+
parser_datasets_version_optional.add_argument(
547+
'-r',
548548
'--dir-mode',
549549
dest='dir_mode',
550550
choices=['skip', 'zip'],

0 commit comments

Comments
 (0)