They may have taken the song from the Punjabi movie but I doubt that they also took the story from the Diljit-movie...
Anyways, the song is - imo - better picturised in the Salman movie...(In Jihne Mera Dil Luteya the song isn't really danced).
Oh wow...so much negativity, driven? Why?