mirror of
				https://github.com/open-metadata/OpenMetadata.git
				synced 2025-10-31 10:39:30 +00:00 
			
		
		
		
	
							parent
							
								
									0edb13fd2c
								
							
						
					
					
						commit
						cf268c9cdd
					
				| @ -94,20 +94,6 @@ class BigquerySource(CommonDbSourceService): | |||||||
|         ) |         ) | ||||||
|         self.temp_credentials = None |         self.temp_credentials = None | ||||||
| 
 | 
 | ||||||
|     #  and "policy_tags" in column and column["policy_tags"] |  | ||||||
|     def prepare(self): |  | ||||||
|         try: |  | ||||||
|             if self.connection_config.enablePolicyTagImport: |  | ||||||
|                 self.metadata.create_tag_category( |  | ||||||
|                     CreateTagCategoryRequest( |  | ||||||
|                         name=self.connection_config.tagCategoryName, |  | ||||||
|                         description="", |  | ||||||
|                         categoryType="Classification", |  | ||||||
|                     ) |  | ||||||
|                 ) |  | ||||||
|         except Exception as err: |  | ||||||
|             logger.error(err) |  | ||||||
| 
 |  | ||||||
|     @classmethod |     @classmethod | ||||||
|     def create(cls, config_dict, metadata_config: OpenMetadataConnection): |     def create(cls, config_dict, metadata_config: OpenMetadataConnection): | ||||||
|         config: WorkflowSource = WorkflowSource.parse_obj(config_dict) |         config: WorkflowSource = WorkflowSource.parse_obj(config_dict) | ||||||
| @ -131,6 +117,18 @@ class BigquerySource(CommonDbSourceService): | |||||||
| 
 | 
 | ||||||
|     def prepare(self): |     def prepare(self): | ||||||
|         self.service_connection.database = self.service_connection.projectId |         self.service_connection.database = self.service_connection.projectId | ||||||
|  |         #  and "policy_tags" in column and column["policy_tags"] | ||||||
|  |         try: | ||||||
|  |             if self.source_config.includeTags: | ||||||
|  |                 self.metadata.create_tag_category( | ||||||
|  |                     CreateTagCategoryRequest( | ||||||
|  |                         name=self.connection_config.tagCategoryName, | ||||||
|  |                         description="", | ||||||
|  |                         categoryType="Classification", | ||||||
|  |                     ) | ||||||
|  |                 ) | ||||||
|  |         except Exception as err: | ||||||
|  |             logger.error(err) | ||||||
|         return super().prepare() |         return super().prepare() | ||||||
| 
 | 
 | ||||||
|     def fetch_sample_data(self, schema: str, table: str) -> Optional[TableData]: |     def fetch_sample_data(self, schema: str, table: str) -> Optional[TableData]: | ||||||
|  | |||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user
	 Mayur Singal
						Mayur Singal