Skip to content

Commit e9de627

Browse files
committed
update
1 parent f35188c commit e9de627

File tree

8 files changed

+439
-110
lines changed

8 files changed

+439
-110
lines changed

dwf.conf

+9-3
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,17 @@ LOG_FILE = run.log
66
LOG_LEVEL = debug
77

88
[DB]
9-
DB_HOST = 192.168.6.106
9+
DB_HOST = 192.168.10.22
1010
DB_PORT = 5432
11-
DB_NAME = develop
11+
DB_NAME = postgres
1212
DB_USER = postgres
13-
DB_PASSWORD = xlearn123
13+
DB_PASSWORD = nA5&Vvkq
14+
15+
#DB_HOST = 192.168.6.106
16+
#DB_PORT = 5432
17+
#DB_NAME = develop
18+
#DB_USER = postgres
19+
#DB_PASSWORD = xlearn123
1420

1521
[TEST_DB]
1622
DB_HOST = 192.168.10.18

dwf/dataset/metadata/dataset_crud.py

+52-42
Original file line numberDiff line numberDiff line change
@@ -8,112 +8,123 @@
88
# Version 0.1
99
#
1010

11-
from dwf.ormmodels import Dataset, datetime
12-
from dwf.common.log import logger
1311
from dwf.common.exception import *
12+
from dwf.common.log import logger
13+
from dwf.ormmodels import Dataset, datetime
1414
from dwf.util.id import generate_primary_key
1515

1616

1717
class DatasetCRUD:
18+
1819
def __init__(self, db_session):
1920
self.db_session = db_session
2021

2122
def add_dataset(self, name, datasource_id, subid=None, creator=None, owner=None, current_process=None,
2223
last_modifier=None, data_file_format=None, default_filter_string=None, description=None,
2324
filter=None, patterns=None, target_entity_class=None):
24-
'''
25-
Register a dataset into the metadata DB.
26-
27-
Args:
28-
name - The name of dataset.
29-
datasource_id - The datasource id of datasource that the dataset belongs to.
30-
data_format - The format of dataset.
31-
data_filter - The filter path of dataset, used for filtering out data from datasource.
32-
data_type - The type of dataset.
25+
"""
26+
创建数据集元信息
3327
34-
Returns:
35-
The ID of added dataset.
36-
'''
28+
:param name: 数据集名称
29+
:param datasource_id: 数据源ID
30+
:param subid:
31+
:param creator:
32+
:param owner:
33+
:param current_process:
34+
:param last_modifier:
35+
:param data_file_format:
36+
:param default_filter_string:
37+
:param description:
38+
:param filter:
39+
:param patterns:
40+
:param target_entity_class:
41+
:return: 数据集ID
42+
"""
3743

3844
id = generate_primary_key('DSET')
3945
create_time = datetime.now()
4046

47+
if creator is None:
48+
creator = 'admin'
49+
if owner is None:
50+
owner = 'admin'
51+
if last_modifier is None:
52+
last_modifier = 'admin'
53+
4154
dataset = Dataset(id=id, subid=subid, creator=creator, owner=owner, current_process=current_process,
4255
last_modifier=last_modifier, create_time=create_time, name=name,
4356
data_file_format=data_file_format, datasource_id=datasource_id,
4457
default_filter_string=default_filter_string, description=description, filter=filter,
4558
patterns=patterns, target_entity_class=target_entity_class)
4659
self.db_session.add(dataset)
4760
self.db_session.commit()
61+
4862
return id
4963

5064
def get_dataset(self, dataset_id):
51-
'''
52-
Get a dataset by ID from the metadata DB of DWF.
53-
54-
Args:
55-
dataset_id - The ID of dataset.
65+
"""
66+
通过ID获取数据集元信息
5667
57-
Returns:
58-
The object of dataset.
68+
:param dataset_id: 数据集ID
69+
:return: 数据集元信息
70+
"""
5971

60-
'''
6172
dataset = self.db_session.query(Dataset).get(dataset_id)
73+
6274
return dataset
6375

6476
def get_all_dataset(self):
65-
'''
66-
Get all datasets from the metadata DB of DWF.
77+
"""
78+
获取全部的数据集元信息
6779
68-
Args:
69-
None
80+
:return: 数据集元信息列表
81+
"""
7082

71-
Returns:
72-
The list of object of dataset.
73-
'''
7483
datasets = self.db_session.query(Dataset).all()
84+
7585
return datasets
7686

7787
def delete_dataset(self, dataset_id):
78-
'''
79-
Delete a dataset by ID from the metadata DB of DWF.
88+
"""
89+
删除数据集元信息
8090
81-
Args:
82-
dataset_id - The ID of dataset.
91+
:param dataset_id: 数据集ID
92+
:return: 无
93+
"""
8394

84-
'''
8595
pending = self.db_session.query(Dataset).get(dataset_id)
8696
self.db_session.delete(pending)
8797
self.db_session.commit()
88-
return True
8998

9099
def update_dataset(self, dataset_id, subid=None, creator=None, owner=None, current_process=None, last_modifier=None,
91100
name=None, datasource_id=None, data_file_format=None, default_filter_string=None,
92101
description=None, filter=None, patterns=None, target_entity_class=None):
93102
"""
103+
更新数据集元信息
94104
95-
:param dataset_id:
105+
:param dataset_id: 数据集ID
96106
:param subid:
97107
:param creator:
98108
:param owner:
99109
:param current_process:
100110
:param last_modifier:
101-
:param name:
102-
:param datasource_id:
111+
:param name: 数据集名称
112+
:param datasource_id: 数据源ID
103113
:param data_file_format:
104114
:param default_filter_string:
105115
:param description:
106116
:param filter:
107117
:param patterns:
108118
:param target_entity_class:
109-
:return:
119+
:return:
110120
"""
111-
pending = self.db_session.query(Dataset).get(dataset_id)
112121

113122
if dataset_id is None:
114-
logger.error('dataset_id is needed')
123+
logger.error('缺少数据集ID')
115124
raise PARAM_LACK
116125

126+
pending = self.db_session.query(Dataset).get(dataset_id)
127+
117128
if subid is not None:
118129
pending.subid = subid
119130
if creator is not None:
@@ -143,4 +154,3 @@ def update_dataset(self, dataset_id, subid=None, creator=None, owner=None, curre
143154

144155
pending.update_time = datetime.now()
145156
self.db_session.commit()
146-
return pending

dwf/datasource/metadata/datasource_crud.py

+73-42
Original file line numberDiff line numberDiff line change
@@ -1,37 +1,60 @@
11
# -*- coding:utf-8 -*-
2+
#
3+
# DataWay Dataset SDK
4+
# Initial Date: 2018.06.14
5+
#
6+
# Title: methods for metadata of datasource
7+
#
8+
# Version 0.1
9+
#
10+
11+
from dwf.common.exception import *
12+
from dwf.common.log import logger
213
from dwf.ormmodels import Datasource, datetime
314
from dwf.util.id import generate_primary_key
415

516

617
class DataSourceCRUD:
18+
719
def __init__(self, db_session):
820
self.db_session = db_session
921

1022
def add_datasource(self, name, subid=None, creator=None, owner=None, current_process=None, last_modifier=None,
1123
data_file_format=None, database_name=None, datasource_type='LOCAL_FS', description=None,
1224
folder_depth=None, paramone=None, password=None, server_ip=None, server_port=None, username=None,
1325
workbench_url=None):
14-
'''
15-
Register a datasource in the metadata DB of DWF.
16-
Args:
17-
id -
18-
create_time -
19-
update_time -
20-
name - Name of DB.
21-
hostname - The hostname or IP of DB.
22-
port - The Port of DB.
23-
username - The username used for login.
24-
password - The password of DB.
25-
datasource_type - The type of datasource, like HDFS.
26-
view_metadata_port -
27-
description - The description of datasource
28-
Returns:
29-
The ID of datasource.
30-
31-
'''
26+
"""
27+
创建数据源元信息
28+
29+
:param name: 数据源名称
30+
:param subid:
31+
:param creator:
32+
:param owner:
33+
:param current_process:
34+
:param last_modifier:
35+
:param data_file_format:
36+
:param database_name:
37+
:param datasource_type:
38+
:param description:
39+
:param folder_depth:
40+
:param paramone:
41+
:param password:
42+
:param server_ip:
43+
:param server_port:
44+
:param username:
45+
:param workbench_url:
46+
:return: 数据源ID
47+
"""
48+
3249
id = generate_primary_key('DSOU')
3350
create_time = datetime.now()
3451

52+
if creator is None:
53+
creator = 'admin'
54+
if owner is None:
55+
owner = 'admin'
56+
if last_modifier is None:
57+
last_modifier = 'admin'
3558
if database_name is None:
3659
database_name = '/'
3760
if folder_depth is None:
@@ -45,49 +68,53 @@ def add_datasource(self, name, subid=None, creator=None, owner=None, current_pro
4568
username=username, workbench_url=workbench_url)
4669
self.db_session.add(new_datasource)
4770
self.db_session.commit()
48-
return new_datasource.id
71+
72+
return id
4973

5074
def get_datasource(self, datasource_id):
51-
'''
52-
Get a datasource by ID from the metadata DB of DWF.
53-
Args:
54-
datasource_id - The ID of datasource.
55-
Returns:
56-
The object of datasource.
57-
'''
75+
"""
76+
根据ID获取数据源元信息
77+
78+
:param datasource_id: 数据源ID
79+
:return: 数据源元信息
80+
"""
81+
5882
datasource = self.db_session.query(Datasource).get(datasource_id)
83+
5984
return datasource
6085

6186
def get_all_datasource(self):
62-
'''
63-
Get all datasources from the metadata DB of DWF.
64-
Args:
87+
"""
88+
获取全部的数据源元信息
89+
90+
:return: 数据源元信息列表
91+
"""
6592

66-
Returns:
67-
The list of datasources.
68-
'''
6993
datasource_list = self.db_session.query(Datasource).all()
94+
7095
return datasource_list
7196

7297
def delete_datasource(self, datasource_id):
73-
'''
74-
Delete a datasource by ID from the metadata DB of DWF.
75-
Args:
76-
datasource_id - The ID of datasource.
77-
'''
98+
"""
99+
删除数据源元信息
100+
101+
:param datasource_id: 数据源ID
102+
:return: 无
103+
"""
104+
78105
pending = self.db_session.query(Datasource).get(datasource_id)
79106
self.db_session.delete(pending)
80107
self.db_session.commit()
81-
return True
82108

83109
def update_datasource(self, datasource_id, name=None, subid=None, creator=None, owner=None, current_process=None,
84110
last_modifier=None, data_file_format=None, database_name=None, datasource_type=None,
85111
description=None, folder_depth=None, paramone=None, password=None, server_ip=None,
86112
server_port=None, username=None, workbench_url=None):
87113
"""
114+
更新数据源元信息
88115
89-
:param datasource_id:
90-
:param name:
116+
:param datasource_id: 数据源ID
117+
:param name: 数据源名称
91118
:param subid:
92119
:param creator:
93120
:param owner:
@@ -104,8 +131,13 @@ def update_datasource(self, datasource_id, name=None, subid=None, creator=None,
104131
:param server_port:
105132
:param username:
106133
:param workbench_url:
107-
:return:
134+
:return:
108135
"""
136+
137+
if datasource_id is None:
138+
logger.error('缺少数据源ID')
139+
raise PARAM_LACK
140+
109141
pending = self.db_session.query(Datasource).get(datasource_id)
110142

111143
if subid is not None:
@@ -145,4 +177,3 @@ def update_datasource(self, datasource_id, name=None, subid=None, creator=None,
145177

146178
pending.update_time = datetime.now()
147179
self.db_session.commit()
148-
return pending

0 commit comments

Comments
 (0)